1.5.1
roboflow/rf-detr1.5.1Feb 27, 2026by Borda
AI Summary
Fixes nested transforms and various model API issues to improve augmentation pipeline reliability.
Key Highlights
- Nested Albumentations transforms (`OneOf`, `Sequential`) now work correctly.
- Dataset transform pipeline now uses torchvision-native `Compose` and `ToImage`.
- Fixed `RFDETRMedium` missing from the public API (`__all__`).
- Fixed supercategory filtering in `_load_classes` for mixed dataset structures.
- Fixed crash in geometric transforms when samples contain zero-area masks.
New Features
- Support for nested Albumentations containers in augmentation pipelines.
Full Release Notes
## 🚀 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)
```python
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](https://www.linkedin.com/in/jirka-borovec/)) – *Nested Albumentations support, transform pipeline refactor, mask fix, supercategory fix, onnxsim pin, CI/testing infrastructure*
---
**Full Changelog**: https://github.com/roboflow/rf-detr/compare/1.5.0...1.5.1