Skip to content

Commit

Permalink
[Fix] make category_id in CocoWholeBodyDataset as ``numpy.arr…
Browse files Browse the repository at this point in the history
…ay`` rather than ``List``. (#2963)
  • Loading branch information
zgjja authored Apr 23, 2024
1 parent de574ed commit b4c5547
Showing 1 changed file with 1 addition and 1 deletion.
Original file line number Diff line number Diff line change
Expand Up @@ -125,7 +125,7 @@ def parse_data_info(self, raw_data_info: dict) -> Optional[dict]:
'segmentation': ann['segmentation'],
'area': area,
'id': ann['id'],
'category_id': ann['category_id'],
'category_id': np.array(ann['category_id']),
# store the raw annotation of the instance
# it is useful for evaluation without providing ann_file
'raw_ann_info': copy.deepcopy(ann),
Expand Down

0 comments on commit b4c5547

Please sign in to comment.