此repo是 Colorization的pytorch实现。 在ECCV中,2016年。
版本信息:pytorch 0.3.0,python 3.5
安装
git clone
培训和验证
从头开始培训:
python main.py
或从指定时间点恢复训练:
python main.py -R ***
以上两个命令默认都支持交叉验证,可以通过添加选项-V False.来关闭交叉验证-V False. 更详细的配置设置,请参考
推理
python main.py -T False -I *** # *** means the inference step.
DAVIS 2017的结果
训练损耗曲线
注意:如您所见,我没有训练足够的时间(因为我没有NVIDIA的支持...)。初始lr为3.16e-4,原始为3.16e-5。
致谢
如代码中所述,该代码的某些部分是从richzha
1