YOLOv5-tf
YOLOv5-tf copied to clipboard
IndexError
Hello,
At first, thank you for making this repository open source, great work. The algorithm seems to work on most of the datasets that I have tried without any problems. However, on one particular dataset(which is bigger but very similar to the others), I get an error,
image, boxes = util.resize(image, boxes) File "/home/usr/yolo/YOLOv5-tf/utils/util.py", line 55, in resize boxes[:, [0, 2]] = boxes[:, [0, 2]] * scale + dw IndexError: too many indices for array: array is 1-dimensional, but 2 were indexed
I cant for the life of me debug this, the shapes seem to be consistent. I understand that this might not be enough information for you to work with but I was just wondering if you have ever had this kind of a problem.
Thanks in advance.
I might suggest original YOLOv5 for getting a better result