v1.5.1: Nested transforms
π Added
-
Nested Albumentations transforms.
OneOfandSequentialcontainers now work correctly inside the augmentation pipeline. Probability settings on container transforms are ignored β they always fire, keeping composition predictable. Inference pipelines can also passNonetargets so the same transform object works for both training and inference. (#752)from rfdetr import RFDETRSmall model = RFDETRSmall() model.train( dataset_dir="...", aug_config={ "OneOf": [ {"RandomBrightnessContrast": {"p": 0.5}}, {"HueSaturationValue": {"p": 0.5}}, ], "HorizontalFlip": {"p": 0.5}, }, )
π± Changed
- Dataset transform pipeline now uses torchvision-native
Compose,ToImage, andToDtypeinstead of custom implementations.Normalizedefaults to ImageNet mean/std. (#745)
π§ Fixed
- Fixed
RFDETRMediummissing from the public API β__all__contained a duplicateRFDETRSmallentry instead. (#748) - Fixed
AR50_90reporting an incorrect value inMetricsMLFlowSinkdue to a wrong COCO evaluation index. (#735) - Fixed supercategory filtering in
_load_classesfor COCO datasets with flat or mixed supercategory structures. (#744) - Fixed a crash in geometric transforms (flip, crop, etc.) when a sample contains zero-area / empty masks. (#727)
- Fixed segmentation training on Colab β
DepthwiseConvBlocknow disables cuDNN for depthwise separable convolutions. (#728) - Locked
onnxsimto<0.6.0to preventpip installfrom hanging indefinitely. (#749)
π Contributors
A special welcome to our new contributors and a big thank you to everyone who helped with this release:
- tillfri (@tillfri) β Fix AR50_90 metric index in MLflow sink
- justin-alt-account (@justin-alt-account) β Fix
RFDETRMediummissing from__all__ - Jirka Borovec (@Borda) (LinkedIn) β Nested Albumentations support, transform pipeline refactor, mask fix, supercategory fix, onnxsim pin, CI/testing infrastructure
Full Changelog: 1.5.0...1.5.1