Skip to content

Conversation

Johansmm
Copy link
Contributor

New alodataset : CocoExtendDetectionDataset. New version of CocoDetectionDataset with the following new features:

  • SplitMixin mode compatibility
  • Include stuff categories from stuff_annotations.json
  • Return multiples labels found in annotations (supercategory, isthing, iscrowd).

Also, update of CocoDetectionDataset to incorporate return multiples labels feat.

Closes #79 #97

@Johansmm Johansmm linked an issue Oct 21, 2021 that may be closed by this pull request
@Johansmm Johansmm requested a review from thibo73800 October 21, 2021 09:30
@Johansmm Johansmm self-assigned this Oct 21, 2021
@Johansmm Johansmm added alodataset enhancement New feature or request labels Oct 21, 2021
from aloscene import Frame


class CocoExtendDetectionDataset(CocoDetectionDataset, SplitMixin):
Copy link
Contributor

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

CocoExtend refer to THE coco dataset right ? Not sure this name refect this property, it seem this is an extension of coco.
What about:

  • CocoBaseDataset: To handle datasets following the coco format
  • CocoDetectionDataset: to handle THE coco dataset (with crowd, things & stuff) ?

Copy link
Contributor Author

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Done

@Johansmm Johansmm requested a review from thibo73800 November 8, 2021 17:04
@thibo73800 thibo73800 merged commit 67acff4 into master Nov 9, 2021
@thibo73800 thibo73800 deleted the 79-coco-detection-splitmixin branch March 10, 2022 09:54
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
alodataset enhancement New feature or request
Projects
None yet
Development

Successfully merging this pull request may close these issues.

CocoDataset for multiples labels Update COCO dataset and COCODetection to detr with SplitMix class
2 participants