颜色分类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
训练流程:
-
训练网络模型:
$ python train.py
-
继续训练:
$ python train.py --continue_train=True
-
测试:
$ python test.py --test_img=test1.jpg
培训结果:
- 目标:参考像素对像素