一段让你合不拢腿的代码 一段让你合不拢的友谊
这篇文章给大家聊聊关于一段让你合不拢腿的代码,用python脱掉人物图片的衣服,以及对应的知识点,希望对各位有所帮助,不要忘了收藏本站哦。
用python脱掉图片人物的衣服
(人要善良,把裤子换成团吧)
环境配置
首先,玩这个模型需要满足以下先决条件:
python3.6版本
有GPU环境(因为源码需要在GPU的Cuda环境下运行,如果没有GPU,建议使用PAI中的DSWhttps://data.aliyun.com/product/learn,比较便宜)
执行以下过程:
下载实验代码(下面会提供下载地址)
下载预先训练的模型。如果只是实验,就用训练好的模型即可(如果只想做换裤子的实验,下载pants2skirt_mhp_instagan模型即可):(下面会提供下载地址)
3)在代码中安装requirement.txt中的依赖包
最后将下载的模型文件放到代码文件的根目录下。目录结构如下(蓝色部分为模型文件夹,包含200_net_G_A.pth和200_net_G_B.pth):
完成以上步骤后,整个环境就搭建完成了。
使用模型转换图像
注意:本项目的代码有很多硬代码逻辑,需要严格遵循才能跑通,包括所有文件的命名。
(1)首先设置要转换的图像
在datasets目录下添加test文件夹,构建如下文件格式
testA和testB存储需要转换的原始图像,类似于“效果演示”中穿裤子的图像,testA_seg和testB_seg需要存储mask图像。 mask文件是转换图中的裤子款式,如下图,对应示例图左边女生的裤子(如果要转换其他部分,就不用一一解释了) )
(2)执行图像转换逻辑
在项目根目录下执行以下代码,使用项目下的test.py测试代码:
如果没有报错,则在GPU环境下1分钟即可完成执行。代码执行后,您可以在项目的结果文件夹中看到转换后的图像。