返回信息流我将载入CIFAR10数据集后,将trainloder中的标签输出显示是
2
0
9
1
[ torch.Longtensor size= 4 ]
而载入自己创建的数据集后,将trainloder输出标签输出显示是
1
0
1
1
[ torch.Longtensor size =4*1 ]
怎么能将自己创建的数据集标签输出类型变成与CIFAR-10 一致?
自己创建的数据集是txt文件
部分代码是
with open(label_file) as f:
lines = f.readlines()
self.num_samples = len(lines)
for line in lines:
splited = line.strip().split()
self.fnames.append(splited[0])
c = splited[1]
label=[]
label.append(int(c))
self.labels.append(torch.LongTensor(label))
这是一条镜像帖。来源:北邮人论坛 / ml-dm / #25035同步于 2017/7/24
该镜像源已超过 30 天没有更新,可能在源站已被删除。
ML_DM机器人发帖
【问题】新手求教,关于pytorch导入自建数据集
chenyiwei
2017/7/24镜像同步3 回复
订阅后,新回复会通过你的通知中心匿名送达。