jingyim / DeepNude_NoWatermark_withModel

DeepNude源码，去水印，带三个模型文件下载地址(cm.lib,mm.lib,mn.lib)，附使用说明，即跑。

DeepNude

DeepNude source code 源代码 去水印 附加模型下载地址

供广大程序员技术交流使用

Prerequisite

Before launch the script install these packages in your Python3 environment:

• numpy
• Pillow
• setuptools
• six
• torch
• torchvision
• wheel

建议使用Conda安装 :)

 conda create -n deepnude -c anaconda python=3.6 numpy Pillow setuptools six pytorch torchvision wheel
 conda activate deepnude

Models

• CLI Checkpoints: 在运行之前需下载三个.lib文件，之后在项目根目录下新建checkpoints目录，将下载的三个文件放至checkpoints目录下

• 备用磁力链接：

magnet:?xt=urn:btih:7BE4EB8D640742D2FFEBD6495E9392E9E2C399BC

若以上Google drive链接挂了可使用该链接。下载速度还蛮快的:)

Launch the script

环境配好，模型下好之后便可以运行代码了！

 python3 main.py

The script will transform input.png to output.png.

GPU

本项目默认使用id为0的GPU运行。

若运行环境不带GPU，则报错。如果没有GPU或想使用CPU运行程序，请将gan.py中

self.gpu_ids = [0] #FIX CPU

改为

self.gpu_ids = [] #FIX CPU

Links

• https://pytorch.org/