Oneflow-Inc/one-yolov5

one-yolov5 v1.2.0 精度测试目标精度

ccssu opened this issue · 1 comments

ccssu commented

yolov5s

后端 metrics/mAP_0.5, metrics/mAP_0.5:0.95,
OneFlow 37.659 56.992
PyTorch 37.4 56.8

全部数据 results.txt
启动指令:

python -m oneflow.distributed.launch --nproc_per_node 4  \ 
train.py --batch-size 128 --data coco.yaml --weights " " --cfg models/yolov5s.yaml --img 640 --epochs 300