DeepMosaics

You can use it to automatically remove the mosaics in images and videos, or add mosaics to them.
This porject based on ‘semantic segmentation’ and ‘Image-to-Image Translation’.

Notice

The code do not include the part of training, I will finish it in my free time.

Prerequisites

• Linux, Mac OS, Windows
• Python 3.6+
• ffmpeg 3.4
• Pytorch 1.0+ (Old version codes)
• CPU or NVIDIA GPU + CUDA CuDNN

Getting Started

Clone this repo:

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

Get pre_trained models and test video

You can download pre_trained models and test video and replace the files in the project.
[Google Drive] [百度云,提取码7thu]

Dependencies

This code depends on opencv-python, available via pip install.

Simple example

• Add Mosaic (output video will save in './result')

python3 deepmosaic.py

• Clean Mosaic (output video will save in './result')

python3 deepmosaic.py --mode clean --model_path ./pretrained_models/clean_hands_unet_128.pth --media_path ./result/hands_test_AddMosaic.mp4

More parameters

If you want to test other image or video, please refer to this file. [options.py]

Acknowledgments

This code borrows heavily from [pytorch-CycleGAN-and-pix2pix] [Pytorch-UNet].