【机器学习】AI去马赛克工具（图像修复）

效果展示

  • 二次元图像（条状遮挡）在这里插入图片描述

  • 二次元图像（带马赛克）
    在这里插入图片描述

  • 三次元图像（条形遮挡）
    在这里插入图片描述

  • 三次元图像（其它遮挡）
    在这里插入图片描述

写在前面

万恶的马赛克：请看下图，本来是一张非常正常的动图，打码后就变得少儿不宜了。今天在github上看到了一个基于深度学习的去马赛克项目，分享给大家。

动图出自番剧《花丸幼稚园》，萌系/日常/治愈类型，非常好看，可以在B站免费追番）

在这里插入图片描述

项目地址

https://github.com/deeppomf/DeepCreamPy

软件下载

https://www.jianguoyun.com/p/DQRUgh4Q26PuBxi_4c4D

项目简介

DeepCreamPy是一个基于深度学习的工具，采用合理的重构方式修复hentai中经过审查（打码）的艺术品。

用前须知

使用前，你必须在Photoshop等图片编辑软件中将包含马赛克的区域用绿色着色。因为DeepCreamPy将绿色着色的图像作为输入，然后神经网络会自动填充打码的区域。

主要功能

  • 去除图像中任意形状的遮挡（如黑线，红心等）
  • 去除图像中的马赛克

注意事项

  • 输入的图片格式必须是png
  • 仅适用于轻微或中等打码，完全打码则无效
  • 着色时选择绿色(0, 255, 0)且要关闭抗锯齿或使用像素笔
  • 二次元图像去码效果更好，三次元图片不太理想（去码是在二次元图像上训练的）

在这里插入图片描述

使用教程

对于条状物

  1. 将图片转为png格式（必须）
  2. 对图片中的条形遮挡采用像素笔进行绿色涂抹
  3. 将第2步保存后的图片复制到decensor_input文件夹中
  4. 双击运行main.exe
  5. 选择Bar censor

对于马赛克

  1. 将带马赛克的图片1转为png格式
  2. 复制一份图片1，副本我们称之为图片2
  3. 对图片2中的马赛克采用像素笔进行绿色涂抹
  4. 将图片2复制到decensor_input文件夹中并重命名为1.png，图片1复制到decensor_input_original文件夹中也重命名为1.png
  5. 双击运行main.exe
  6. 选择Mosaic censor

温馨提示

没有关闭抗锯齿的后果：
在这里插入图片描述
对于Windows系统，推荐使用电脑自带的3D画图软件。画笔选择像素笔，颜色设置为(0, 255, 0)，然后画笔大小根据需要调整：
在这里插入图片描述

拓展阅读

1、TecoGAN——去除薄码

https://github.com/thunil/TecoGAN

在这里插入图片描述
2、deep-image-prior——图像修复和超分

https://github.com/DmitryUlyanov/deep-image-prior

在这里插入图片描述

posted @ 2020-11-29 22:33  XavierJ  阅读(6845)  评论(0)    收藏  举报