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

浙公网安备 33010602011771号