DeepMosaics

Дипмозаики

Английский | 中文
Вы можете использовать его для автоматического удаления мозаики в изображениях и видео или добавить к ним мозаику.
Этот проект основан на «семантической сегментации» и «переводе изображения на изображение».
Попробуйте на этом сайте!

Примеры

источник	Автоматическое добавление мозаики	Авто чистая мозаика

• По сравнению с Deepcreampy

Мозаичный образ	Deepcreampy	наш

• Стиль перенос

источник	Ван Гог	к зиме

Интересный пример: Рикардо Милос для кошки

Запустить DeepMosaics

Вы можете запустить DeepMosaics через предварительно построенный двоичный пакет или из Source.

Попробуйте в Интернете

Вы можете просто попытаться удалить мозаику на лице на этом сайте.

Предварительно построенный двоичный пакет

Для Windows мы выбиваем версию GUI для легкого тестирования.
Загрузите эту версию и предварительно обученную модель через [Google Drive] [百度云, 提取码 1x0a]

• [Справочный документ]
  
• Video Turning => [youtube] [bilibili]

Внимание:

• Требуется Windows_x86_64, Windows10 лучше.
  
• Различные предварительно обученные модели подходят для различных эффектов. [Введение в предварительно обученные модели]
  
• Время выполнения зависит от производительности компьютеров (версия GPU имеет лучшую производительность, но требует установки CUDA).
  
• Если выходное видео не может быть воспроизведено, вы можете попробовать с PotPlayer.
  
• Обновления версии GUI медленнее, чем источник.
  

Бежать от источника

Предварительные условия

• Linux, Mac OS, Windows
• Python 3.6+
• FFMPEG 3.4.6
• Pytorch 1.0+
• CPU или NVIDIA GPU + CUDA CUDNN
  

Зависимости

Этот код зависит от OpenCV-Python, Torchvision, доступной через установку PIP.

Клонировать это репо

git clone https://github.com/HypoX64/DeepMosaics.git
cd DeepMosaics

Получите предварительно обученные модели

Вы можете скачать предварительные модели и поместить их в './pretriand_models'.
[Google Drive] [百度云, 提取码 1x0a]
[Введение в предварительно обученные модели]

Чтобы добавить/удалить мозаику, должна быть модельный файл mosaic_position.pth at ./pretrained_models/mosaic/mosaic_position.pth

Установить зависимости

(Необязательно) Создайте виртуальную среду

virtualenv mosaic
source mosaic/bin/activate

Затем установите зависимости

pip install -r requirements.txt

Если вы не можете построить scikit-image , запуск export CFLAGS='-Wno-implicit-function-declaration затем попробуйте восстановить.

Простой пример

• Добавить Mosaic (выходные носители будут сохранены в ./Result ')
  

python deepmosaic.py --media_path ./imgs/ruoruo.jpg --model_path ./pretrained_models/mosaic/add_face.pth --gpu_id 0

• Чистая мозаика (выходной носитель будет сэкономить в./Result ')
  

python deepmosaic.py --media_path ./result/ruoruo_add.jpg --model_path ./pretrained_models/mosaic/clean_face_HD.pth --gpu_id 0

Если вы видите ошибку Please check mosaic_position_model_path! , Проверьте, есть ли модель -файл с именем mosaic_position.pth at ./pretrained_models/mosaic/mosaic_position.pth

Больше параметров

Если вы хотите проверить другие изображения или видео, пожалуйста, обратитесь к этому файлу.
[options_introduction.md]

Обучение с собственным набором данных

Если вы хотите тренироваться со своим собственным набором данных, пожалуйста, см.

Благодарности

Этот код в значительной степени заимствует [pytorch-cyclegan-and-pix2pix] [pytorch-unet] [pix2pixhd] [bisenet] [dfdnet] [gfrnet_pytorch_new].