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".
origin | auto add mosaic | auto clean mosaic |
---|---|---|
mosaic image | DeepCreamPy | ours |
---|---|---|
origin | to Van Gogh | to winter |
---|---|---|
An interesting example:Ricardo Milos to cat
You can either run DeepMosaics via pre-built binary package or from source.
For windows, we bulid a GUI version for easy test.
Download this version and pre-trained model via [Google Drive] [百度云,提取码1x0a]
Attentions:
This code depends on opencv-python, torchvision available via pip install.
git clone https://github.com/HypoX64/DeepMosaics
cd DeepMosaics
You can download pre_trained models and put them into './pretrained_models'.
[Google Drive] [百度云,提取码1x0a]
[Introduction to pre-trained models]
python3 deepmosaic.py --media_path ./imgs/ruoruo.jpg --model_path ./pretrained_models/mosaic/add_face.pth --use_gpu -1
python3 deepmosaic.py --media_path ./result/ruoruo_add.jpg --model_path ./pretrained_models/mosaic/clean_face_HD.pth --use_gpu -1
If you want to test other image or video, please refer to this file.
[options_introduction.md]
If you want to train with your own dataset, please refer to training_with_your_own_dataset.md
This code borrows heavily from [pytorch-CycleGAN-and-pix2pix] [Pytorch-UNet] [pix2pixHD] [BiSeNet].