浏览代码

index fix

Dimitri Korsch 6 年之前
父节点
当前提交
0cfb0f6b24
共有 1 个文件被更改,包括 3 次插入0 次删除
  1. 3 0
      nabirds/annotations/inat.py

+ 3 - 0
nabirds/annotations/inat.py

@@ -59,6 +59,9 @@ class INAT19_Annotations(BaseAnnotations):
 		# load only if present
 		if self.has_parts:
 			super(INAT19_Annotations, self)._load_parts()
+			# set part idxs from 1-idxs to 0-idxs
+			self.part_locs[..., 0] -= 1
+
 		self._load_bounding_boxes()
 
 	def _load_split(self):