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

时间:2024-02-20 22:13:47

效果展示

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

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

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

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

写在前面

万恶的马赛克:请看下图,本来是一张非常正常的动图,打码后就变得少儿不宜了。今天在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

在这里插入图片描述