-
Notifications
You must be signed in to change notification settings - Fork 1.8k
-
Notifications
You must be signed in to change notification settings - Fork 1.8k
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
自己数据集训练的yolov9模型tensorrt的输出结果与pytorch结果差别很大,精度完全没有对齐 #1535
Comments
请问有人训练自己的数据集后pytorch和tensorrt的结果对齐了吗 |
@guanshanjushib 试试用官方的模型 |
你能发一下模型的结构吗,在pytorch下print打印就可以。如果和官网的一致,可以考虑提高一下tensorrt的版本,采用trt 8.6。 |
yolov9_uav76_20240521_key.txt |
要是有兴趣,我们可以考虑实现一下SyncBatchNorm。email: [email protected] |
可以啊,你们能实现SyncBatchNorm嘛?另外,我要是用onnx转成engine,会不会也会因为SyncBatchNorm出现精度问题? |
|
我测试了一下,wts转换出来的engine运行后,精度差异很大,用onnx转engine,运行后精度损失很少 |
Env
About this repo
master-branch
yolov9
Your problem
我用yolov9算法训练自己的数据集,得到的模型经过yolov9的tensorrtx推理,发现精度对不齐,pytorch输出的结果与tensorrt输出的结果差别很大,有些目标框的位置和精度没对齐。请问当前的yolov9推理对齐了吗?
The text was updated successfully, but these errors were encountered: