v0.15.1

[0.15.1] - 2026-05-28

Added

• Add image logging for all fine-tuning tasks. Sample predictions are saved locally and
  forwarded to configured loggers (TensorBoard, Weights & Biases, MLflow).
• Add predict_batch for batched inference across all fine-tuning tasks.
• Configurable patch_size for DINOv3 LT-DETR models.
• Added decoder and losses from D-FINE.
• Add support for choosing LR scheduler for LTDETR object detection. You can specify the
  scheduler with model_args.scheduler_name, choosing either linear or flat-cosine.
• ONNX export for models that support dynamic batch sizes can now specify if the batch
  size should be dynamic with the dynamic_batch_size keyword argument.
• Explicit support for all YOLO26 variants.

Fixed

• Fix PicoDet fine-tuning with mismatched num_classes.
• Fix DINOv3 LT-DETR patch size precedence so model_args.patch_size overrides the
  backbone default.

What's Changed

• Cleanup augmentation update by @yutong-xiang-97 in #705
• Simon add coco instance segmentation dataset by @simonschoelly in #687
• Add Jinja2 template for object detection by @simonschoelly in #708
• Fix Training Crash Caused by Assertion Failure in generalized_box_iou by @yutong-xiang-97 in #707
• Better hash for data mmap hash by @simonschoelly in #706
• Object Detection Image Logging (Open for Discussion) by @mrpositron in #709
• Fix default font loading to handle TypeError gracefully by @mrpositron in #713
• added LightlyTrain in Research section by @gabrielfruet in #716
• Add image logging for classification training by @mrpositron in #714
• Fix DINOv3 patch size precedence by @gabrielfruet in #717
• Fix PicoDet finetune with mismatched num_classes by @mrpositron in #721
• Add image logging for semantic segmentation training by @mrpositron in #718
• Decode RLE encoded COCO segmetation datasets by @simonschoelly in #710
• Add image logging for instance segmentation training by @mrpositron in #722
• Add image logging for panoptic segmentation training by @mrpositron in #724
• Add Flat Cosine Scheduler for LT-DETR by @gabrielfruet in #719
• Add gemma4_vit_ architecture support to TIMM model prefixes by @mrpositron in #727
• Add D-FINE Decoder by @yutong-xiang-97 in #711
• Fix dynamic batch sizes for onnx export by @simonschoelly in #591
• Dynamic batch size in changelog by @simonschoelly in #732
• Refactor training visualizations into task-specific handlers by @mrpositron in #726
• Feat support variable patch size for DINOv3 LT-DETR by @gabrielfruet in #720
• Image logging support for picodet by @mrpositron in #733
• Add yolo26 models by @liopeer in #735
• Transition to uv sync by @liopeer in #730
• Adds image logging to ML tracking platforms by @mrpositron in #736
• Add Predict Batch Method to Object Detection by @mrpositron in #738
• Add Predict Batch Method to Instance Segmentation by @mrpositron in #741
• Add Predict Batch Method to Image Classification by @mrpositron in #746
• add resolve auto for scheduler params by @yutong-xiang-97 in #740
• Refactor predict to reuse predict_batch components in object detection by @mrpositron in #748
• Add Predict Batch Method to Semantic Segmentation by @mrpositron in #744
• Add Predict Batch Method to Panoptic Segmentation by @mrpositron in #747
• Preparing for release 0.15.1 by @mrpositron in #754

Full Changelog: v0.15.0...v0.15.1