Skip to content

Conversation

@bernardev254
Copy link

Summary

This PR updates the augmentations() function in hackathon/objectdetection.py to support bounding box and label augmentation using create_global_crops.

Changes

  • Passes bboxes, labels, and num_classes into create_global_crops.
  • Receives and handles the transformed bounding boxes and labels (as crops_heatmaps).
  • Stores the augmented object data in the returned dictionary under the "objects" key.
  • Keeps local_crops logic optional and unchanged.

Motivation

This enables consistent transformation of object detection targets (bounding boxes and labels) during augmentation, aligning with the changes made to create_global_crops and preparing the pipeline for training with augmented samples.
closes #3

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.

Complete 'augmentations' function for bounding box pipeline

1 participant