Skip to content

v1.5.1: Nested transforms

Choose a tag to compare

@Borda Borda released this 27 Feb 10:22
· 578 commits to develop since this release

πŸš€ Added

  • Nested Albumentations transforms. OneOf and Sequential containers now work correctly inside the augmentation pipeline. Probability settings on container transforms are ignored β€” they always fire, keeping composition predictable. Inference pipelines can also pass None targets 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, and ToDtype instead of custom implementations. Normalize defaults to ImageNet mean/std. (#745)

πŸ”§ Fixed

  • Fixed RFDETRMedium missing from the public API β€” __all__ contained a duplicate RFDETRSmall entry instead. (#748)
  • Fixed AR50_90 reporting an incorrect value in MetricsMLFlowSink due to a wrong COCO evaluation index. (#735)
  • Fixed supercategory filtering in _load_classes for 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 β€” DepthwiseConvBlock now disables cuDNN for depthwise separable convolutions. (#728)
  • Locked onnxsim to <0.6.0 to prevent pip install from 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 RFDETRMedium missing 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