README.md

image

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

More example

origin auto add mosaic auto clean mosaic
image image image
image image image
mosaic image DeepCreamPy ours
image image image
image image image
  • Style Transfer
origin to Van Gogh to winter
image image image

An interesting example:Ricardo Milos to cat

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 and pre-trained model via [Google Drive] [百度云,提取码1x0a]

image
Attentions:

  • Require Windows_x86_64, Windows10 is better.
  • Different pre-trained models are suitable for different effects.[Introduction to pre-trained models]
  • Run time depends on computer performance(The current version does not support gpu, if you need to use gpu please run source).
  • If output video cannot be played, you can try with potplayer.
  • GUI version update slower than source.

Run from source

Prerequisites

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

You can download pre_trained models and put them into './pretrained_models'.
[Google Drive] [百度云,提取码1x0a]
[Introduction to pre-trained models]

Simple example

  • Add Mosaic (output media will save in './result')
python3 deepmosaic.py --media_path ./imgs/ruoruo.jpg --model_path ./pretrained_models/mosaic/add_face.pth --use_gpu 0
  • Clean Mosaic (output media will save in './result')
python3 deepmosaic.py --media_path ./result/ruoruo_add.jpg --model_path ./pretrained_models/mosaic/clean_face_HD.pth --use_gpu 0

More parameters

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

Training with your own dataset

If you want to train with your own dataset, please refer to training_with_your_own_dataset.md

Acknowledgments

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

项目简介

使用深度学习方法去掉图片或视频中的马赛克

发行版本

当前项目没有发行版本

贡献者 2

HypoX64 @weixin_36721459
H HypoX64 @HypoX64

开发语言

  • Python 100.0 %