11 2020 档案

摘要:在深度学习中 我们首先要准备数据集,尤其时用pytorch进行模型搭建训练时,需要首先构建dataset,然后封装为dataloder。 在生成dataset时,我们通过图片路径读取图片,然后进行一系列图片变换,生成可用于训练的dataset,我们需要把图片路径保存在一个.txt文件中, 通过读取. 阅读全文
posted @ 2020-11-17 00:09 learningcaiji 阅读(1114) 评论(0) 推荐(0)
摘要:利用pytorch 定义自己的网络模型时,需要继承toch.nn.Module 基类。 基类中有parameters()、modules()、children()等方法 import torch import torch.nn as nn class myModel(nn.Module): def 阅读全文
posted @ 2020-11-03 22:03 learningcaiji 阅读(793) 评论(0) 推荐(0)