stack expects each tensor to be equal size, but got [3, 224, 224] at entry 0 and [1,224,224] at entr
首先可以确定地是在数据读取的以及处理发生的问题。目前可以归纳为一下两种:1:使用Image.open()没有添加.convert('RGB')错误代码例子:class Mydataset(data.Dataset): def __init__(self,root,labels,transform): super(Mydataset,self).__init__() self.imgs_path = root self.labels
原创
2021-07-23 14:36:26 ·
10777 阅读 ·
6 评论