-
Notifications
You must be signed in to change notification settings - Fork 413
New issue
Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.
Already on GitHub? Sign in to your account
在COCO上直接训练v5lite-s 416x416,未修改任何参数,map仅35.2 #63
Comments
直接使用v5封好的test.py文件进行评测要比正常的低1-2个点,请使用scripts脚本里面的eval.py评价脚本,直接调用cocotools的api进行评测:https://github.com/ppogg/YOLOv5-Lite/blob/master/scripts/eval.py,使用方式为,test.py里面生产的xx_precisions.json替换脚本中的路径: |
你好~~~ 这是使用你提供的v5lite-s.pt模型eval.py评价脚本评测的结果: Average Precision (AP) @[ IoU=0.50:0.95 | area= all | maxDets=100 ] = 0.228 这是我直接训练v5lite-s的模型,eval.py评价脚本评测结果: |
大佬,其实我想请教一下,为什么使用原始参数,直接训练v5lite-s模型,怎么比你的低2个点,需要做调参?还是需要使用预训练的backbone? |
还是太低了,如果相差只有0.1或者0.2个百分点是正常的,但是[email protected]已经相差了四五个百分点了,你测一下v5lite-g.pt这个模型,然后待会也贴一下指标,谢谢! |
v5lite-g这个模型你提供的没有下载下来,我训练一个,再来贴。请问训练的时候还有什么需要注意的地方? 期待回复,谢谢! |
|
这种训练策略针对v5lite-s、v5lite-c、v5lite-g均奏效吗? |
对于v5lite-s凑效,对于c和g,马赛克的scale需要相应变成0.2,0.5,在最后的15个epoch |
朋友你好,其他两个模型的评测指标还未发,另外,你需要先解决在你的评价指标和仓库模型测试相差了四五个点的问题,这样方便你在后续复现。你可以加我qq1138099162,这几天晚上帮你把coco api的评估流程跑一遍。 |
啊,为啥下不下来??? |
谢谢你持续的回复与帮助,直接用你的模型v5lite-s评测,这几个点的差距,主要是我把conf_thresh由0.0001改为0.1,iou由0.45改为0.5,修改回去能得到你公布的mAP。 我加你QQ了,请同意一下,谢谢。 |
好 |
直接用原始参数在COCO上训练v5lite-s,输入416x416,测试的结果如下:
模型测试命令:python test.py --device 0 --conf-thres 0.1 --iou-thres 0.5
Class Images Labels P R [email protected] [email protected]:.95: 100%|█| 79/79 [00:45<00:00
all 5000 36335 0.537 0.363 0.352 0.203
使用博主提供的模型v5lite-s,输入416x416,测试的结果如下:
模型测试命令:python test.py --device 0 --conf-thres 0.1 --iou-thres 0.5
Class Images Labels P R [email protected] [email protected]:.95: 100%|█| 79/79 [00:48<00:00
all 5000 36335 0.542 0.388 0.373 0.225
mAP相差2个点,请问这是什么原因导致的呢?期待大佬的回复!谢谢。
The text was updated successfully, but these errors were encountered: