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

fixed the random rotation augmentation inconsistency #155

Merged

Conversation

ChristopherKotthoff
Copy link
Contributor

Update Augmentation Pipeline to Improve Coverage and Efficiency

Description of Changes

  • Updated RandomRotation to cover a full 360-degree range (angle=[0, 360], sample_style="range"), replacing the previous fixed [90, 90] which provided no rotation augmentation.
  • Adjusted RandomFlip probability from 0.4 to 0.5 for a simpler and more balanced augmentation strategy.
  • Simplified flipping to use a single RandomFlip with horizontal=True instead of separate horizontal and vertical flips.

Justification

  1. Rotation Augmentation: The previous configuration with [90, 90] effectively applied no rotation variation. The updated 360-degree range ensures diverse rotations and better augmentation coverage.
  2. Flip Probability Change: A probability of 0.5 aligns with the standard approach, ensuring equal chances of flipping or not flipping, which simplifies implementation and interpretation.
  3. Single Flip Augmentation: A single horizontal flip is sufficient to achieve all unique mirroring transformations when combined with 360-degree rotations. Adding a separate vertical flip would be redundant.

Impact

This change improves the diversity and efficiency of the augmentation pipeline, ensuring a better-distributed dataset for training.

@PatBall1 PatBall1 merged commit c983eda into PatBall1:master Nov 22, 2024
1 of 2 checks passed
@ChristopherKotthoff ChristopherKotthoff deleted the chris/rotation-augmentation-fix branch November 22, 2024 23:57
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.

2 participants