颜色分类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培训结果目标参考像素对像素
下载地址
用户评论