sczhou/CodeFormer

Что это?

Ключевые возможности

• 01Восстановление лиц вслепую на изображениях и обрезанных, выровненных лицах размером 512x512
• 02Настраиваемый fidelity weight для баланса между качеством и сходством с исходным изображением
• 03Обработка видео в форматах .mp4, .mov и .avi
• 04Раскрашивание чёрно-белых и выцветших фотографий лиц
• 05Заполнение замаскированных областей на лицах
• 06Дополнительное улучшение фона с помощью Real-ESRGAN и увеличение разрешения лиц

Сценарии использования

• Восстановление старых семейных фотографий или снимков с низким разрешением
• Улучшение размытых кадров из старых видеозаписей
• Исправление искажённых лиц в изображениях, созданных ИИ
• Сравнение методов восстановления лиц вслепую на наборах выровненных изображений лиц

Быстрый старт

1. 1

   Клонируйте репозиторий

   git clone https://github.com/sczhou/CodeFormer
2. 2

   Установите зависимости Python

   pip3 install -r requirements.txt
3. 3

   Скачайте предварительно обученные модели CodeFormer

   python scripts/download_pretrained_models.py CodeFormer
4. 4

   Улучшите изображение целиком

   python inference_codeformer.py -w 0.7 --input_path [image folder]|[image path]

Что важно знать

• !Проект распространяется по специальной лицензии NTU S-Lab License 1.0, которая не является стандартной лицензией с открытым исходным кодом. Перед повторным использованием ознакомьтесь с её условиями.
• !Для запуска необходимы PyTorch и CUDA, а также загрузка нескольких предварительно обученных моделей.

Темы

Частые вопросы

Что такое CodeFormer?

Инструмент на Python для восстановления лиц вслепую: улучшает размытые, низкокачественные и повреждённые лица на фото и видео с помощью трансформера с поиском по кодовой книге. CodeFormer восстанавливает лица на изображениях и видео, когда тип ухудшения неизвестен. Этот подход называется восстановлением лиц вслепую. Проект — официальная реализация статьи, представленной на NeurIPS 2022: модель использует обученную кодовую книгу априорных признаков лиц и трансформер, чтобы восстановить чёткие детали по изображению низкого качества.

CodeFormer можно использовать бесплатно? Какая у него лицензия?

CodeFormer распространяется с открытым исходным кодом под лицензией Other. Перед использованием в коммерческих продуктах проверьте условия лицензии.

CodeFormer всё ещё поддерживается?

Нет коммитов больше 6 месяцев. Последний коммит в CodeFormer — 18 нояб. 2025 г..

Для чего можно использовать CodeFormer?

Восстановление старых семейных фотографий или снимков с низким разрешением Улучшение размытых кадров из старых видеозаписей Исправление искажённых лиц в изображениях, созданных ИИ Сравнение методов восстановления лиц вслепую на наборах выровненных изображений лиц

Как установить CodeFormer?

Клонируйте репозиторий: git clone https://github.com/sczhou/CodeFormer Установите зависимости Python: pip3 install -r requirements.txt Скачайте предварительно обученные модели CodeFormer: python scripts/download_pretrained_models.py CodeFormer Улучшите изображение целиком: python inference_codeformer.py -w 0.7 --input_path [image folder]|[image path]

Где скачать CodeFormer?

Исходный код и инструкции по установке — на GitHub: https://github.com/sczhou/CodeFormer