๐ Summary
The v8.3.7 release introduces several enhancements, with a focus on fixing argument handling for custom predictors in YOLO, along with updates for performance and stability improvements.
๐ Key Changes
- Custom Predictor Argument Fix: Corrected the usage of arguments in
model.predict()
to support custom predictors. - Docker Image Update: The base Docker image is now upgraded to utilize PyTorch 2.4.1, with CUDA 12.1 and cuDNN 9.
- New Script for Synthetic Datasets: Added a function to create synthetic COCO datasets, supporting data testing and augmentation.
- Enhanced AutoBatch Memory Management: Improved GPU memory handling during autobatching to optimize resource usage.
- Added
OMP_NUM_THREADS=1
: Adjusted Docker configurations for improved CPU management and performance.
๐ฏ Purpose & Impact
- ๐ Improved Custom Predictor Functionality: Ensures smoother operation for users implementing custom predictors, providing accurate argument handling on initialization.
- ๐ Performance Boost: The Docker update incorporates the latest improvements in PyTorch, enhancing model training and inference speed.
- ๐ Simplified Data Handling: The synthetic COCO dataset script facilitates the creation of testing datasets, helping developers easily prepare data for model validation.
- ๐งน Efficient Resource Management: Changes in autobatch memory handling and Docker configuration aim to reduce memory overhead and CPU saturation, ensuring better performance on a variety of systems.
- ๐ Streamlined Development: These updates collectively enhance the developer experience by optimizing workflows, reducing potential bugs, and improving code clarity.
What's Changed
- FROM pytorch/pytorch:2.4.1-cuda12.1-cudnn9-runtime by @glenn-jocher in #16725
- Fix absolute stylesheet CSS path by @glenn-jocher in #16726
- New
create_synthetic_coco_dataset
function by @UltralyticsAssistant in #16742 - Added YOLO11 to Explorer models by @ambitious-octopus in #16739
- Fix NVIDIA Typo by @lakshanthad in #16728
- setuptools>=70.0.0 fix CVE-2024-6345โ by @glenn-jocher in #16727
- AutoBatch improve cache clearing by @UltralyticsAssistant in #16744
- Adding
OMP_NUM_THREADS=1
by @glenn-jocher in #16747 ultralytics 8.3.7
custom Predictor args fix by @Rick-v-E in #16734
Full Changelog: v8.3.6...v8.3.7