We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent 3716ff4 commit 0b16c7cCopy full SHA for 0b16c7c
1 file changed
efficientdet/dataset.py
@@ -73,7 +73,7 @@ def load_annotations(self, image_index):
73
74
annotation = np.zeros((1, 5))
75
annotation[0, :4] = a['bbox']
76
- annotation[0, 4] = a['category_id']
+ annotation[0, 4] = a['category_id'] - 1
77
annotations = np.append(annotations, annotation, axis=0)
78
79
# transform from [x, y, w, h] to [x1, y1, x2, y2]
0 commit comments