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