yjh0410 93b7481820 optimize yolov7 codes преди 2 години
..
README.md a7d44da089 update README преди 2 години
build.py 3f8c6ecea7 add ONNX deployment преди 2 години
loss.py 311a9b89b7 fix a unknown bug in YOLOX преди 2 години
matcher.py 2727b6adb1 debug trainer преди 2 години
yolov7.py 93b7481820 optimize yolov7 codes преди 2 години
yolov7_backbone.py 93b7481820 optimize yolov7 codes преди 2 години
yolov7_basic.py 93b7481820 optimize yolov7 codes преди 2 години
yolov7_head.py 7cf531e7da add Tracking преди 2 години
yolov7_neck.py 7cf531e7da add Tracking преди 2 години
yolov7_pafpn.py 93b7481820 optimize yolov7 codes преди 2 години

README.md

YOLOv4:

  • For training, we train YOLOv7 and YOLOv7-Tiny with 300 epochs on COCO.
  • For data augmentation, we use the large scale jitter (LSJ), Mosaic augmentation and Mixup augmentation, following the setting of YOLOv5.
  • For optimizer, we use SGD with momentum 0.937, weight decay 0.0005 and base lr 0.01.
  • For learning rate scheduler, we use linear decay scheduler.
  • For YOLOv7's structure, we use decoupled head, following the setting of YOLOX.
  • While YOLOv7 incorporates several technical details, such as anchor box, SimOTA, AuxiliaryHead, and RepConv, I found it too challenging to fully reproduce. Instead, I created a simpler version of YOLOv7 using an anchor-free structure and SimOTA. As a result, my reproduction had poor performance due to the absence of the other technical details. However, since it was only intended as a tutorial, I am not too concerned about this gap.
Model Backbone Batch Scale APval
0.5:0.95
APval
0.5
FLOPs
(G)
Params
(M)
Weight
YOLOv7-Tiny ELANNet-Tiny 1xb16 640 38.0 56.8 22.6 7.9 ckpt
YOLOv7 ELANNet-Large 1xb16 640 48.0 67.5 144.6 44.0 ckpt