生成绘画火炬
根据作者的,对PyTorch重新。
先决条件
该代码已经在Ubuntu 14.04上进行了测试,以下是需要安装的主要组件:
Python3
PyTorch 1.0+
火炬视觉0.2.0+
张量板
pyyaml
训练模型
python train.py --config configs/config.yaml
检查点和日志将保存到checkpoints 。
用训练好的模型进行测试
默认情况下,它将在检查点中加载最新保存的模型。 您也可以使用--iter通过迭代选择保存的模型。
训练有素的PyTorch模型:[ ] []
python test_single.py \
--image examples/imagenet/imagenet_patches_ILSVRC2012_val_00008210_input.png \
--mask examples/cen
1