Jacobi93 / Alpha-IoU

[NeurIPS 2021] Alpha-IoU: A Family of Power Intersection over Union Losses for Bounding Box Regression
GNU General Public License v3.0
154 stars 21 forks source link

yolov5x,yolov5s weights on COCO #1

Closed jamjamjon closed 2 years ago

jamjamjon commented 2 years ago

would you mind upload yolov5s, yolov5x weights trained on COCO? Thx.

Jacobi93 commented 2 years ago

Thx so much for your interest in our repo. We did experiments on COCO with YOLOv5 version 3.0 (https://github.com/ultralytics/yolov5/releases), which is different from this repo, i.e., YOLOv5 version 4.0. YOLOv5 has been updated for many versions. Plz re-train them using either this repo or apply our method to the latest version of YOLOv5 (https://github.com/ultralytics/yolov5). You should obtain close results. Thx.