litchi99 / DeepMosaics

Automatically remove the mosaics in images and videos, or add mosaics to them.

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

Notes

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

Prerequisites

• Linux, (I didn't try this code on Windows or mac machine)
• Python 3.5+
• ffmpeg
• Pytroch 0.4, (I will update to 1.0)
• 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] [百度云,提取码z8vz]

Dependencies

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

AddMosaic

python3 AddMosaic.py

CleanMosaic

copy the AddMosaic video from './result' to './video_or_image'

python3 CleanMosaic.py

More parameters

[addmosaic_options] [cleanmosaic_options]

Acknowledgments

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