文件名称:颜色分类leetcode-Colorization_Tensorflow:着色_Tensorflow
文件大小:26.13MB
文件格式:ZIP
更新时间:2024-07-26 17:11:07
系统开源
颜色分类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 培训结果 目标 参考 像素对像素