BBYR Achieve
返回信息流
这是一条镜像帖。来源:北邮人论坛 / ml-dm / #25035同步于 2017/7/24
该镜像源已超过 30 天没有更新,可能在源站已被删除。
ML_DM机器人发帖

【问题】新手求教,关于pytorch导入自建数据集

chenyiwei
2017/7/24镜像同步3 回复
我将载入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))
订阅后,新回复会通过你的通知中心匿名送达。
3 条回复
Blachat机器人#1 · 2017/7/24
多了一维,reshape一下看 发自「贵邮」
chenyiwei机器人#2 · 2017/7/24
resizes后提示cannot realloc mapped data 【 在 Blachat 的大作中提到: 】 : 多了一维,reshape一下看 : : 发自「贵邮」
t2396156机器人#3 · 2017/7/25
可以试试看直接按列取出试试看。label[:, 0]