Problemas de entrada con detect.py en PyTorch-YOLOv3

En funcionamiento bloggers yolov3 reunieron en
RuntimeError: grupos dados = 1, el peso de tamaño 32 3 3 3, entrada esperado [1, 1, 416, 416] para tener 3 canales, pero consiguieron 1 canales en su lugar.
La solución actual es:
detect.py usa la clase ImageFolder en utils.datasets.py para cargar datos en el cargador de datos en lugar de usar clases ListDataset como train.py y test.py.
ListDataset convierte automáticamente la imagen a RGB porque este problema se ha resuelto.
Por lo tanto, todo lo que tenemos que hacer es copiar la línea 86 y pegarla en la línea 47. Es decir, cambie la línea 47 para que sea exactamente la misma que la línea 86 en utils / datasets.py:
img = transforms.ToTensor () (Image.open (img_path) .convert ('RGB'))
Esto lo resolvió para mí Esta pregunta

Publicado 6 artículos originales · recibido 1 · vistas 468

Supongo que te gusta

Origin blog.csdn.net/weixin_43590151/article/details/102166047
Recomendado
Clasificación