PyTorch-YOLOv3中detect.py 出现的输入问题

博主在运行yolov3中遇见
RuntimeError: Given groups=1, weight of size 32 3 3 3, expected input[1, 1, 416, 416] to have 3 channels, but got 1 channels instead。
现解决方法为：
detect.py使用utils.datasets.py中的ImageFolder类将数据加载到数据加载器中，而不是使用诸如train.py和test.py之类的ListDataset类。
ListDataset自动将图像转换为RGB，因为此问题已解决。
因此，我们要做的是复制第86行并将其粘贴到第47行。也就是说，将第47行更改为与utils / datasets.py中的第86行完全相同：
img = transforms.ToTensor（）（Image.open（img_path）.convert（‘RGB’））
这为我解决了这个问题。