Implement improved zoom augmentations through albumentations module #1089
Add this suggestion to a batch that can be applied as a single commit.
This suggestion is invalid because no changes were made to the code.
Suggestions cannot be applied while the pull request is closed.
Suggestions cannot be applied while viewing a subset of changes.
Only one suggestion per line can be applied in a batch.
Add this suggestion to a batch that can be applied as a single commit.
Applying suggestions on deleted lines is not supported.
You must change the existing code in this line in order to create a valid suggestion.
Outdated suggestions cannot be applied.
This suggestion has been applied or marked resolved.
Suggestions cannot be applied from pending reviews.
Suggestions cannot be applied on multi-line comments.
Suggestions cannot be applied while the pull request is queued to merge.
Suggestion cannot be applied right now. Please check back later.
This PR implements a comprehensive, modular augmentation system to address the core challenge of generalizing across sensors and acquisition conditions in airborne biodiversity observation. The new system moves augmentations from inline implementation to a dedicated module with configurable zoom augmentations for improved multi-resolution training.
Key Features
🔧 Modular Augmentation System
src/deepforest/augmentations.py
module with 10+ augmentationsdatasets/training.py
to reusable module🔍 Zoom Augmentations for Multi-Resolution Training
Implements the specifically requested augmentations:
scale_range
parameter)⚙️ Flexible Configuration Options
🔄 Full Backward Compatibility
augment=True
) still usesHorizontalFlip
transforms
parameterExample Usage for Multi-Resolution Training
Changes Made
src/deepforest/augmentations.py
with configurable transform systemaugmentations
field totrain
section in config.yamlBoxDataset
to accept augmentations configurationtrain_dataloader()
to pass config-based augmentationsTesting
Benefits for Airborne Biodiversity Observation
This implementation directly addresses the stated challenge by providing:
Fixes #735.
Warning
Firewall rules blocked me from connecting to one or more addresses
I tried to connect to the following addresses, but was blocked by firewall rules:
huggingface.co
python -m pytest tests/test_main.py -k train -x -v
(dns block)If you need me to access, download, or install something from one of these locations, you can either:
💡 You can make Copilot smarter by setting up custom instructions, customizing its development environment and configuring Model Context Protocol (MCP) servers. Learn more Copilot coding agent tips in the docs.