Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

added registries #1862

Merged

Conversation

shaydeci
Copy link
Contributor

Added registreis for the coco and yolo format detection classes.
Once registered users can simply plug them into their yaml files if they have the right file structure without the need to specify an entire datalloader (like we do in our main config i.e stating coco2017_train...):

train_dataloader_params:
  dataset: COCOFormatDetectionDataset
  batch_size: 25
  num_workers: 8
  shuffle: True
  drop_last: True
  pin_memory: True
  collate_fn: DetectionCollateFN

Copy link
Collaborator

@ofrimasad ofrimasad left a comment

Choose a reason for hiding this comment

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

LGTM

Copy link
Contributor

@BloodAxe BloodAxe left a comment

Choose a reason for hiding this comment

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

LGTM

@ofrimasad ofrimasad merged commit 8819667 into master Feb 26, 2024
9 checks passed
@ofrimasad ofrimasad deleted the feature/SG-000_register_coco_format_and_yolo_datasets branch February 26, 2024 11:28
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

Successfully merging this pull request may close these issues.

3 participants