search

tlpss / keypoint-detection

2D keypoint detection with Pytorch Lightning and wandb
MIT License
70 stars 9 forks source link

Fixed a bug where multiple category ids could not be obtained #15

Closed canhe173 closed 1 year ago

canhe173 commented 1 year ago

Changing category = category_dict[category.id] to category = category_dict[annotation.category_id] in coco_dataset.py 128th line.

tlpss commented 1 year ago

fixes #14