ColorFlow: мультяшная затененность, автоматическая затененность черно-белых изображений для улучшения согласованности цветов и качества изображения

Общее введение

ColorFlow - это инструмент автоокрашивания последовательностей изображений, разработанный командой Tencent's ARC для решения проблемы автоокрашивания черно-белых последовательностей изображений. ColorFlow использует трехфазный механизм моделирования диффузии в сочетании с механизмом самовнушения для достижения высокого качества затенения изображения и согласованности цветов. Этот инструмент особенно подходит для промышленного применения в сериях мультфильмов и комиксов, значительно улучшая управляемость и идентичность окраски изображений.

Список функций

• Извлечь улучшенные трубки для раскрашивания: Поиск и подбор цвета по пулу эталонных изображений для обеспечения согласованности цветов.
• Контекстная окраска труб: Раскрашивание изображений с использованием контекстной информации для сохранения идентичности персонажей и объектов.
• Направленный трубопровод сверхвысокого разрешения: Повышает разрешение изображения, улучшает его детализацию и качество.
• Механизм самовнушения: Использование механизмов самовнимания в диффузионных моделях для надежного контекстного обучения и подбора цветов.
• Бенчмарки ColorFlow-Bench: Обеспечивает всесторонний эталонный сравнительный анализ для оценки производительности модели.

Использование помощи

Процесс установки

1. склад клонов::

   git clone https://github.com/TencentARC/ColorFlow
cd ColorFlow

2. Настройка среды Python: Убедитесь, что Anaconda или Miniconda установлены, затем создайте и активируйте среду Python и установите необходимые зависимости:

   conda create -n colorflow python=3.8.5
conda activate colorflow
pip install -r requirements.txt

3. Запуск приложения: Запустите интерфейс Gradio:

   python app.py

4. Доступ к ColorFlow: Откройте http://localhost:7860 в браузере. Если вы запускаете приложение на удаленном сервере, замените localhost на IP-адрес или доменное имя сервера.

Функции Поток операций

1. Извлечь улучшенную раскраску::
   • Загрузите пул эталонных изображений, и ColorFlow автоматически найдет и совместит цвета.
   • Выберите последовательность черно-белых изображений, которые необходимо раскрасить, и система подберет цвет и раскрасит их в соответствии с пулом эталонных изображений.
2. контекстно-зависимая раскраска::
   • Система автоматически анализирует контекстную информацию последовательности изображений, чтобы обеспечить последовательную идентификацию персонажей и объектов.
   • Пользователи могут вручную настроить параметры подбора цвета, чтобы получить наилучшие результаты затенения.
3. Суперразрешение Bootstrap::
   • После завершения окрашивания система автоматически выполняет обработку с суперразрешением для повышения детализации и качества изображения.
   • Пользователи могут выбирать различные уровни суперразрешения, чтобы настроить качество изображения в соответствии со своими потребностями.

Подробное руководство по эксплуатации

• Загрузите эталонное изображение: Нажмите кнопку "Загрузить", выберите файл эталонного изображения, и система автоматически выполнит поиск цвета.
• Выбор последовательности черно-белых изображений: Нажмите на кнопку "Выбрать файл", чтобы загрузить последовательность черно-белых изображений для раскрашивания.
• Настройка параметров цветоподбораВ интерфейсе "Настройка параметров" пользователи могут настроить соответствующие параметры согласования цветов, такие как интенсивность цвета, точность согласования и т. д.
• Просмотр результатов раскрашиванияНажмите кнопку "Начать окрашивание", и система автоматически выполнит процесс окрашивания, а пользователь сможет просмотреть результаты окрашивания в режиме реального времени.
• Сохранить изображение для раскрашивания: По окончании раскрашивания нажмите кнопку "Сохранить", чтобы сохранить раскрашенное изображение в локальном режиме.