search

Gatedip / GDIP-Yolo

Gated Differentiable Image Processing (GDIP) for Object Detection in Adverse Conditions | Accepted at ICRA 2023
http://gatedip.github.io/
MIT License
53 stars 5 forks source link

IndexError #12

Open lovemuyao opened 11 months ago

lovemuyao commented 11 months ago

Traceback (most recent call last): File "D:\Users\pythonProject\GDIP-Yolo-main\GDIP-Yolo-main\utils\datasets.py", line 187, in File "D:\ProgramFiles\anaconda3\envs\pytorch1.12\lib\site-packages\torch\utils\data_utils\fetch.py", line 49, in fetch data = [self.dataset[idx] for idx in possibly_batched_index] File "D:\ProgramFiles\anaconda3\envs\pytorch1.12\lib\site-packages\torch\utils\data_utils\fetch.py", line 49, in data = [self.dataset[idx] for idx in possibly_batched_index] File "D:\Users\pythonProject\GDIP-Yolo-main\GDIP-Yolo-main\utils\datasets.py", line 32, in getitem img_org, bboxes_org = self.__parse_annotation(self.annotations[item]) File "D:\Users\pythonProject\GDIP-Yolo-main\GDIP-Yolo-main\utils\datasets.py", line 81, in parse_annotation img, bboxes = dataAug.RandomHorizontalFilp()(np.copy(img), np.copy(bboxes)) File "..\utils\data_augment.py", line 16, in call bboxes[:, [0, 2]] = w_img - bboxes[:, [2, 0]]

IndexError: too many indices for array: array is 1-dimensional, but 2 were indexed