소스 검색

added L1_pred and L1_full parts to birdsnap

Dimitri Korsch 5 년 전
부모
커밋
72f5e11c85
2개의 변경된 파일20개의 추가작업 그리고 4개의 파일을 삭제
  1. 4 4
      cvdatasets/annotations/impl/birdsnap.py
  2. 16 0
      scripts/info_files/info.yml

+ 4 - 4
cvdatasets/annotations/impl/birdsnap.py

@@ -21,8 +21,8 @@ class BSNAP_Annotations(BBoxMixin, PartsMixin, BaseAnnotations):
 			split_file="tr_ID.txt",
 			# bounding_boxes="bounding_boxes.txt",
 			bounding_box_dtype=np.dtype([(v, np.int32) for v in "xywh"]),
-			# parts_file=join("parts", "part_locs.txt"),
-			# part_names_file=join("parts", "parts.txt"),
+			parts_file=join("parts", "part_locs.txt"),
+			part_names_file=join("parts", "parts.txt"),
 
 		)
 
@@ -30,8 +30,8 @@ class BSNAP_Annotations(BBoxMixin, PartsMixin, BaseAnnotations):
 			[info.images_file, "_images"],
 			[info.labels_file, "labels"],
 			[info.split_file, "_split"],
-			# [info.parts_file, "_part_locs"],
-			# [info.part_names_file, "_part_names"],
+			[info.parts_file, "_part_locs"],
+			[info.part_names_file, "_part_names"],
 			# [info.bounding_boxes, "_bounding_boxes"],
 		]
 		return info

+ 16 - 0
scripts/info_files/info.yml

@@ -352,6 +352,22 @@ PARTS:
     rescale_size: !!int 427
 
 
+  ####################################
+  # BIRDSNAP
+  ####################################
+
+  BIRDSNAP_L1_pred:
+    <<: *birdsnap
+    <<: *parts_l1p
+    annotations: L1_pred
+    rescale_size: !!int 427
+
+  BIRDSNAP_L1_full:
+    <<: *birdsnap
+    <<: *parts_l1f
+    annotations: L1_full
+    rescale_size: !!int 427
+
   ####################################
   # Stanford Cars
   ####################################