聽了bilibil小土堆的課，拿螞蟻蜜蜂數(shù)據(jù)集練手，記錄下自己犯的錯(cuò)~

1、trans = transform.Compose()

# 這段報(bào)錯(cuò)了…… #圖片轉(zhuǎn)為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

哈哈，我看得懂個(gè)錘子……

然后找半天發(fā)現(xiàn)……是多打了一個(gè)torchvision？還是有點(diǎn)不懂

# 這樣又不報(bào)錯(cuò)了 # 圖片轉(zhuǎn)為tensor 修改維度 trans = torchvision.transforms.Compose([transforms.Resize((32, 32)),transforms.ToTensor()])

2、__len__ 不要忘記定義

當(dāng)我沒有定義len的時(shí)候，

train_dataset = ants_dataset + bees_dataset會(huì)報(bào)錯(cuò) def __len__(self):return len(self.img_names)

3、圖像通道改變

image = image.convert("RGB") # png圖片是4通道的，轉(zhuǎn)化為3通道

總結(jié)

以上是生活随笔為你收集整理的初学者之蚂蚁蜜蜂分类报错记录的全部?jī)?nèi)容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網(wǎng)站內(nèi)容還不錯(cuò)，歡迎將生活随笔推薦給好友。