今天我们就利用Python脚本实现天气查询应用吧。直接开整~
从大量照片/卡通数据中习得照片到卡通画的映射。

python版本: 3.6
相关模块:
pytorch 1.4
tensorflow-gpu 1.14
face-alignment
dlib
训练数据包括真实照片和卡通画像,为降低训练复杂度,该项目对两类数据进行了如下预处理:
项目开源了204张处理后的卡通画数据,您还需准备约1000张人像照片(为匹配卡通数据,尽量使用亚洲年轻女性照片,人脸大小最好超过200x200像素),使用以下命令进行预处理:
python data_process.py --data_path YourPhotoFolderPath --save_path YourSaveFolderPath
将处理后的数据按照以下层级存放,trainA、testA中存放照片头像数据,trainB、testB中存放卡通头像数据。
├── dataset└── photo2cartoon├── trainA├── xxx.jpg├── yyy.png└── ...├── trainB├── zzz.jpg├── www.png└── ...├── testA├── aaa.jpg├── bbb.png└── ...└── testB├── ccc.jpg├── ddd.png└── ...
python train.py --dataset photo2cartoon
加载预训练参数:
python train.py --dataset photo2cartoon --pretrained_weights models/photo2cartoon_weights.pt
将一张测试照片(亚洲年轻女性)转换为卡通风格:
python test.py --photo_path ./images/photo_test.jpg --save_path ./images/cartoon_res
文章到这里就结束了,感谢你的观看,如果你也喜欢本次的内容,那就动动小手点赞收藏吧。
上一篇:React 学习笔记:事件处理
下一篇:决策树-相关作业