pytorch学习
pytorch一、训练模型的流程1.定义数据集对象2.搭建网络结构3.训练模型二、pytorch 加载模型方法一:将整个网络都都保存下来方法二:仅保存和加载模型参数三、遇到的问题1.不能成功读取图片一、训练模型的流程1.定义数据集对象__init__(self,......):可以视为构造函数__len__(self):返回读取文件的长度__getitem__(self, index):读取文件指定index多张的图片(输入图、答案);对图片进行预处理操作(裁剪、翻转、归一化等);cv2读取图片
原创
2021-06-24 11:17:22 ·
294 阅读 ·
0 评论