Skip to content

v2.2.0 - YOLOv5 in PyTorch
Compare
Choose a tag to compare
@zhang-dut zhang-dut released this 19 May 01:46
· 33 commits to main since this release
v2.2.0
fd866d7

The main body of this project comes from v7.0 - YOLOv5 SOTA Realtime Instance Segmentation of ultralytics/yolov5.
This version adds ECA attention mechanism in v2.0.0.

Important Updates

  • Add ECA attention mechanism to this network.🎉NEW
  • Using VOC2007 dataset for training.

Specific Changes

  • Modify "models/common_eca.py"🎉NEW
    • Add Class: ECAAttention.
  • Modify "models/yolo.py"🎉NEW
    • Find parse_model function(line 303), adding ECAAttention(line 322).
  • Modify "train.py"
    • line 441: initial weights path -> weights/yolov5s.pt
    • line 442: model.yaml path -> models/parts_yolov5s_eca.yaml🎉NEW
    • line 443: dataset.yaml path -> data/parts_voc2007.yaml

Notes

There are errors in "models/parts_yolov5s_eca.yaml", which will be modified in subsequent versions.⚠️ERROR

  • layer 17: CoordAttention -> ECAAttention
  • layer 21: CoordAttention -> ECAAttention
  • layer 25: CoordAttention -> ECAAttention
Assets 2
Loading