-
Notifications
You must be signed in to change notification settings - Fork 523
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
How about yolov4-p6-light in 640 size? #5
Comments
Calculation of Precision and Recall are different, yolov5 calculate average score of 0.5:0.95 and yolor calculate score of 0.5. |
If that your explain is true, I think yolov4-p6-light.yaml (without reOrg and IDetect module) can not catch the yolov4-csp.yaml results in VisDrone dataset, although yolov4-p6-light.yaml have four output, the same infer time as yolov4-csp.yaml. I'm confuse about it. thank you for your explain! |
In my experiments. maybe for 640 case, p5 models will get better performance than p6 models, but i have not tested it yet. |
thank you so much for your reply, I will test it in my code soon. |
Zigars,我可以加你的QQ或者微信吗?我是一个学生,我想用自己的数据集跑作者的YOLOR,但是没跑通,可以请教下您吗? |
I have test trained yolor-p6.yaml In VisDrone dataset (a famous UAV dataset) use 640 size and your pretrained .pt. And I get a excellent results, It's a great work!
Then, I trained yolov4-p6-light.yaml(remove the reOrg and IDetect module and I fix it in yolov5's rep) in VisDrone dataset use 640 size without pretrained, but the results under the original yolov4-csp‘s result. maybe it have some bug in my code.
So I'm training the same yolov4-p6-light.yaml(remove the reOrg and IDetect module and I fix it in your yolor's rep) in VisDrone use 640 size without pretrained, and I need find the bug in my own code if your yolor rep can get a good results. If not, that says yolov4-p6-light need a coco pretrained? because the model have four output, the loss can hard to convergence?
The all experiment trained in 300epoch, 32batchsize, 640*640 size, use single V100 to train the model. maybe you can solve my question. thks!
The text was updated successfully, but these errors were encountered: