上传者: 42107561
|
上传时间: 2021-10-02 14:51:40
|
文件大小: 25.1MB
|
文件类型: -
快速的神经风格转换
Pytorch实现快速神经样式转换,结合3篇论文:
,
神奈川大浪
春天的街角
用法
火车
如果您只想快速预览一些预训练的样式,请跳过
下载COCO 2014数据集(),假设您放入~/Data/COCO/train2014
准备样式图像,例如assets/styles/bxphai.jpg
基础训练:
python fnst/train.py --dataset ~/Data/COCO/ --style assets/styles/bxphai.jpg
如果您想调整超参数和批处理大小,时期等的数量。
python fnst/train.py --help
程式化
您可以自己训练或使用一些预先训练的样式:
神奈川大浪( checkpoints/wave.pth )
春天的街角checkpoints/dnson.pth Ngoc Son ( checkp