This release aggregates various minor bug fixes and performance improvements since the main v3.0 release and incorporates PyTorch 1.7.0 compatibility updates. v3.1 models share weights with v3.0 models but contain minor module updates (inplace
fields for nn.Hardswish() activations) for native PyTorch 1.7.0 compatibility. For PyTorch 1.7.0 release updates see https://github.com/pytorch/pytorch/releases/tag/v1.7.0.
Breaking Changes
- 'giou' hyperparameter has been renamed to 'box' to better reflect a criteria-agnostic regression loss term (#1120)
Bug Fixes
- PyTorch 1.7 compatibility update.
torch>=1.6.0
required,torch>=1.7.0
recommended (#1233) - GhostConv module bug fix (#1176)
- Rectangular padding min stride bug fix from 64 to 32 (#1165)
- Mosaic4 bug fix (#1021)
- Logging directory runs/exp bug fix (#978)
- Various additional
Added Functionality
- PyTorch Hub functionality with YOLOv5 .autoshape() method added (#1210)
- Autolabelling addition and standardization across detect.py and test.py (#1182)
- Precision-Recall Curve automatic plotting when testing (#1107)
- Self-host VOC dataset for more reliable access and faster downloading (#1077)
- Adding option to output autolabel confidence with --save-conf in test.py and detect.py (#994)
- Google App Engine deployment option (#964)
- Infinite Dataloader for faster training (#876)
- Various additional