pytorch-YOLOv4
pytorch-YOLOv4 copied to clipboard
error:OpenCV can't augment image: 608 x 608
ValueError: could not broadcast input array from shape (348,376,3) into shape (370,376,3)
OpenCV can't augment image: 608 x 608
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
in which file this hsv = list(hsv) change is to be made
@ShivaThe I added after this line and error solved. https://github.com/Tianxiaomo/pytorch-YOLOv4/blob/master/dataset.py#L140