贪玩美女

文章
9
资源
0
加入时间
3年1月8天

pytorch 自定义数据读取方式

pytorch 自定义数据读取方式2一个文件夹下面各个类别的图像数据都有,同时用一个对应的标签文件,比如txt文件来维护图像和标签的对应关系from __future__ import print_function, divisionimport torchimport torch.nn as nnimport torch.optim as optimfrom torch.optim import lr_schedulerfrom torch.autograd import Variable