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.

Run DeepMosaics

You can either run DeepMosaics via pre-built binary package or from source.

Pre-built binary package

For windows, we bulid a GUI version for easy test.
Download this version via [Google Drive] [百度云,提取码1x0a]

Attentions:

Require Windows_x86_64, Windows10 is better.
Run time depends on computer performance.
If output video cannot be played, you can try with potplayer.

Run from source

Prerequisites

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

Dependencies

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

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]

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].

53756e4c69/DeepMosaics