张量流
概述
这是Tensorflow实现,用于 。
要下载培训/测试所需的数据,请参阅数据目录中的README.md。
档案文件
VDSR.py:主要培训文件。
MODEL.py:模型定义。
MODEL_FACTORIZED.py:因式分解CNN的模型定义。 (不建议使用。仅用于记录目的)
PSNR.py:定义如何在python中计算PSNR
TEST.py:测试所有保存的检查点
PLOT.py:从TEST.py绘制测试结果
如何使用
训练
# if start from scratch
python VDSR.py
# if start with a checkpoint
python VDSR.py --model_path ./checkpoints/CHECKPOINT_NAME.ckpt
测验
# this will test all the checkpoint
1