初学者之蚂蚁蜜蜂分类报错记录

听了bilibil小土堆的课，拿蚂蚁蜜蜂数据集练手，记录下自己犯的错~

1、trans = transform.Compose()

# 这段报错了……
#图片转为tensor 修改维度
trans = torchvision.transforms.Compose([torchvision.transforms.Resize((32, 32)),
                                                torchvision.transforms.ToTensor()])

显示的是：stack expects each tensor to be equal size, but got [3, 32, 32] at entry 0 and [1, 32, 32] at entry

哈哈，我看得懂个锤子……

然后找半天发现……是多打了一个torchvision？还是有点不懂

# 这样又不报错了
# 图片转为tensor 修改维度 
trans = torchvision.transforms.Compose([transforms.Resize((32, 32)),
                                                transforms.ToTensor()])

2、__len__ 不要忘记定义

当我没有定义len的时候，

train_dataset = ants_dataset + bees_dataset会报错

    def __len__(self):
        return len(self.img_names)

3、图像通道改变

image = image.convert("RGB") # png图片是4通道的，转化为3通道