search

Tianxiaomo / pytorch-YOLOv4

PyTorch ,ONNX and TensorRT implementation of YOLOv4
Apache License 2.0
4.46k stars 1.49k forks source link

error：OpenCV can't augment image: 608 x 608 #539

Open FYM1209 opened 2 years ago

FYM1209 commented 2 years ago

ValueError: could not broadcast input array from shape (348,376,3) into shape (370,376,3)

OpenCV can't augment image: 608 x 608

pavlik-tikhomirov commented 1 year ago

I had the same issue when trying to train tiny yolov4 model and this post helped me. I added hsv = list(hsv) and the error vanished

ShivaThe commented 1 year ago

in which file this hsv = list(hsv) change is to be made

ozora-ogino commented 1 year ago

@ShivaThe I added after this line and error solved. https://github.com/Tianxiaomo/pytorch-YOLOv4/blob/master/dataset.py#L140