|
@@ -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
|