Skip to content

Conversation

@fcdl94
Copy link
Member

@fcdl94 fcdl94 commented Oct 3, 2025

  • Introduced MosaicAugmentation class to combine images in a grid layout.
  • Added CopyPasteAugmentation class to enhance images by pasting objects from other images.
  • Updated DictDataset methods to support new augmentation functionalities.
  • Refactored existing methods for clarity and consistency, including deprecating the store_coco_roboflow_format method in favor of save.
  • Improved annotation handling and bounding box adjustments in the dataset processing.

Note

Introduce mosaic and copy-paste augmentations with a CLI to generate/save augmented datasets, and update DictDataset to use a new save method with COCO export fixes.

  • Augmentations (scripts/dataset_augmentations.py):
    • Add MosaicAugmentation and CopyPasteAugmentation with bbox remapping and optional blending.
    • Provide CLI to generate augmented images (weights, sizes, counts), write images to disk, build a new dataset, and save in Roboflow COCO format; optional val copy.
  • Dataset (focoos/data/datasets/dict_dataset.py):
    • Add save(output_dir) for Roboflow COCO export; deprecate store_coco_roboflow_format in favor of save.
    • COCO export tweaks: category_id offset by +1; default iscrowd to 0.
    • Minor refactor: typing import and small formatting in from_segmentation.

Written by Cursor Bugbot for commit d85453a. This will update automatically on new commits. Configure here.

…et enhancement

- Introduced MosaicAugmentation class to combine images in a grid layout.
- Added CopyPasteAugmentation class to enhance images by pasting objects from other images.
- Updated DictDataset methods to support new augmentation functionalities.
- Refactored existing methods for clarity and consistency, including deprecating the store_coco_roboflow_format method in favor of save.
- Improved annotation handling and bounding box adjustments in the dataset processing.
@github-actions
Copy link

github-actions bot commented Oct 3, 2025

Coverage

Tests Skipped Failures Errors Time
228 0 💤 0 ❌ 0 🔥 35.135s ⏱️

@fcdl94
Copy link
Member Author

fcdl94 commented Oct 3, 2025

Need to add it to CLI and document it better.

@CuriousDolphin
Copy link
Member

LGTM!

Copy link

@cursor cursor bot left a comment

Choose a reason for hiding this comment

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

This PR is being reviewed by Cursor Bugbot

Details

Your team is on the Bugbot Free tier. On this plan, Bugbot will review limited PRs each billing cycle for each member of your team.

To receive Bugbot reviews on all of your PRs, visit the Cursor dashboard to activate Pro and start your 14-day free trial.

Co-authored-by: ivan.murabito <ivan.murabito@focoos.ai>
@cursor
Copy link

cursor bot commented Dec 10, 2025

Cursor Agent can help with this pull request. Just @cursor in comments and I'll start working on changes in this branch.
Learn more about Cursor Agents

Co-authored-by: ivan.murabito <ivan.murabito@focoos.ai>
@cursor
Copy link

cursor bot commented Dec 10, 2025

Cursor Agent can help with this pull request. Just @cursor in comments and I'll start working on changes in this branch.
Learn more about Cursor Agents

@CuriousDolphin CuriousDolphin merged commit d55ac4e into main Dec 10, 2025
10 checks passed
@CuriousDolphin CuriousDolphin deleted the feat/dataset-augmentation branch December 10, 2025 10:11
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.

4 participants