【文件属性】：
文件名称：Colorization:彩色图像着色的pythorch实现。 在ECCV中，2016年
文件大小：3.85MB
文件格式：ZIP
更新时间：2021-05-09 08:36:11
python pytorch-implmention colorful-image-colorization Python
此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
【文件预览】：
Colorization-master
----utils()
--------img_transforms.py(8KB)
--------frame_utils.py(595B)
--------__init__.py(0B)
--------tools.py(8KB)
--------__pycache__()
----config.py(1KB)
----web_imgs()
--------gt_3.png(809KB)
--------pred_2.png(718KB)
--------lr.png(6KB)
--------pred_3.png(776KB)
--------train_loss.png(17KB)
--------gt_1.png(428KB)
--------gt_2.png(733KB)
--------pred_1.png(428KB)
----models()
--------custom_layers()
--------__init__.py(0B)
--------__pycache__()
----datasets.py(3KB)
----main.py(2KB)
----model.py(6KB)
----LICENSE(1KB)
----losses.py(492B)
----pretrained()
--------download_vgg19_bn.sh(63B)
----readme.md(2KB)
----solver.py(8KB)