RuntimeError: output with shape [1, 300, 300] doesn‘t match the broadcast shape [3, 300, 300]

原因

数据集中存在了灰色图片，导致其维度是1维的，应将其变为RGB，3维的图片。

方法

去文件里找到这个图片`

修改一下：

在这里插入代码片from PIL import Image
import os
import cv2
from torchvision import transforms
path=r'C:\Users\Administrator\Desktop\YOLO\VOCdevkit\VOC2007\JPEGImages\\'

listt=os.listdir(path)


""""
for name in listt:
    #I=Image.open(path+name)
    I1=cv2.imread(path+name,cv2.IMREAD_UNCHANGED)
    try:
        a=I1.shape[2]
    except:
        print(name,I1.shape)
"""
I=cv2.imread(path+'000112.jpg')

print(I.shape)

cv2.imwrite(path+'000112.jpg',I)
print(I.shape)

或者用添加链接描述他的方法