颜色分类leetcode-Colorization_Tensorflow:着色_Tensorflow

颜色分类leetcode 让有颜色的 Tensorflow 2016 年发表的论文的 Tensorflow 实现。 由于缺乏计算能力，一些设置比论文中描述的要小。 有什么不同 无分类网络 输入大小减少到 63x63（较大的输入会导致 OOM 错误） conv2d_transpose 用于着色网络 数据集 验证集用作训练数据（训练数据太大，无法处理） 文件夹设置 -data -training -img1.jpg -img2.jpg -... -validation -val1.jpg -val2.jpg -... 要求 Python 2.7 TensorFlow 1.3 简历2 网络模型 训练 $ python train.py 继续训练 $ python train.py --continue_train=True 测试 $ python test.py --test_img=test1.jpg 培训结果 目标 参考 像素对像素