文章目录
1 简介
主要文件
utils_dataset.py
工具文件,主要用来制作dataset,便于加入dataloader,用于实现数据集的加载和并行读取
model.py
主要写入网络(模型)
train.py
主要用于训练
use.py
加载训练好的模型,用于测试或使用
2 网络设计与开发
1前期
网络结构(调研网络结构,使用的卷积层、上下采样层;in/out_channel、kernel_size。参考github、论文)
数据集(匹配网络、大致处理方法)
2 步骤
0. 图像数据处理
灰度-RGB,上下采样,尺寸一致
注意事项:shape相同,便于设置大于1的batch_size
1. dataset制作 utils_dataset.py
注意事项:
- 读取的图像改为
dtype=np.float32
类型,pytorch网络中参数格式为float32,须保持一致