Skip to content

v3.1 - Bug Fixes and Performance Improvements

Compare
Choose a tag to compare
@glenn-jocher glenn-jocher released this 29 Oct 15:07
· 2186 commits to master since this release
c8c5ef3

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