10-SRCNN-使用CNN实现超分辨成像
文章目录
- utils_dataset.py
- model.py
- train.py
- use.py
主要文件
utils_dataset.py
工具文件，主要用来制作dataset，便于加入dataloader，用于实现数据集的加载和并行读取
model.py
主要写入网络（模型）
train.py
主要用于训练
use.py
加载训练好的模型，用于测试或使用
utils_dataset.py
可以参考上一篇博客9-1-Dataset创建-同文件
'''
1默认：没有改变尺寸，数据集中的图像可以是任意形状尺寸。dataloader中必须令batch_size=1
transforms.Resize((宽,高))(image)