颜色分类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
培训结果
目标
参考
像素对像素
