RuntimeError: output with shape [1, 300, 300] doesn‘t match the broadcast shape [3, 300, 300]
原因数据集中存在了灰色图片,导致其维度是1维的,应将其变为RGB,3维的图片。方法去文件里找到这个图片`修改一下:import osimport cv2from torchvision import transformspath=r’C:\Users\Administrator\Desktop\YOLO\VOCdevkit\VOC2007\JPEGImages\’listt=os.listdir(path)“”""for name in listt:#I=Image.open(pa
原创
2021-06-09 16:13:50 ·
1896 阅读 ·
8 评论