소스 검색

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):