10-SRCNN-使用CNN实现超分辨成像-CSDN博客

阿里云国内75折 回扣 微信号:monov8
阿里云国际,腾讯云国际,低至75折。AWS 93折 免费开户实名账号 代冲值 优惠多多 微信号:monov8 飞机:@monov6

文章目录


主要文件
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) 
阿里云国内75折 回扣 微信号:monov8
阿里云国际,腾讯云国际,低至75折。AWS 93折 免费开户实名账号 代冲值 优惠多多 微信号:monov8 飞机:@monov6