颜色分类leetcode Colorization Tensorflow:着色 Tensorflow

newjorchen 2 0 zip 2024-10-05 13:10:13

颜色分类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

培训结果:

  • 目标:参考像素对像素

用户评论
请输入评论内容
评分:
暂无评论