DeepMosaics

DeepMosaics

英语| 中文
您可以使用它在图像和视频中自动删除马赛克，或向其添加马赛克。
该项目基于“语义细分”和“图像到图像翻译”。
在此网站上尝试！

例子

起源	自动添加马赛克	自动清洁马赛克

• 与DeepCreampy相比

马赛克图像	deepcreampy	我们的

• 样式转移

起源	给梵高	到冬天

一个有趣的例子：里卡多·米洛斯（Ricardo Milos）

运行DeepMosaics

您可以通过预先构建的二进制软件包或来源运行DeepMosaics。

在网上尝试

您只需尝试在此网站上删除脸部的镶嵌物即可。

预先建造的二进制包

对于Windows，我们将GUI版本用于易于测试。
下载此版本，以及通过[Google Drive] [百度云，提取码1x0A]的预训练模型

• [帮助文件]
  
• 视频教程=> [YouTube] [Bilibili]

注意：

• 需要Windows_x86_64，Windows10更好。
  
• 不同的预训练模型适合不同的效果。[介绍预训练模型]
  
• 运行时间取决于计算机性能（GPU版本具有更好的性能，但需要安装CUDA）。
  
• 如果无法播放输出视频，则可以尝试使用Potplayer。
  
• GUI版本的更新比来源慢。
  

从源运行

先决条件

• Linux，Mac OS，Windows
• Python 3.6+
• FFMPEG 3.4.6
• Pytorch 1.0+
• CPU或NVIDIA GPU + CUDA CUDNN
  

依赖性

该代码取决于OpenCV-Python，通过PIP安装可用的火炬。

克隆这个仓库

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

获取预训练的模型

您可以下载pre_trained模型并将其放入'./pretained_models'中。
[Google Drive] [百度云，提取码1x0A]
[预训练模型简介]

为了添加/删除马赛克，必须有一个模型文件mosaic_position.pth at ./pretrained_models/mosaic/mosaic_position.pth

安装依赖项

（可选）创建虚拟环境

virtualenv mosaic
source mosaic/bin/activate

然后安装依赖项

pip install -r requirements.txt

如果您无法构建scikit-image ，则运行export CFLAGS='-Wno-implicit-function-declaration然后尝试重建。

简单示例

• 添加马赛克（输出媒体将保存在./ result'中）
  

python deepmosaic.py --media_path ./imgs/ruoruo.jpg --model_path ./pretrained_models/mosaic/add_face.pth --gpu_id 0

• 清洁马赛克（输出媒体将节省'./ result'）
  

python deepmosaic.py --media_path ./result/ruoruo_add.jpg --model_path ./pretrained_models/mosaic/clean_face_HD.pth --gpu_id 0

如果看到错误Please check mosaic_position_model_path! ，检查是否有一个名为mosaic_position.pth的模型文件./pretrained_models/mosaic/mosaic_position.pth

更多参数

如果要测试其他图像或视频，请参阅此文件。
[options_introduction.md]

使用您自己的数据集进行培训

如果您想使用自己的数据集进行培训，请参阅triending_with_your_own_own_dataset.md

致谢

该代码从[pytorch-cyclegan and-pix2pix] [pytorch-unet] [pix2pixhd] [bisenet] [dfdnet] [gfrnet_pytorch_new] [pytorch-unet] [pytorch-unet] [pytorch-unet]。