Skip to content
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

vit分类模型,与pt模型相比精度掉点太大 #1575

Open
BaseJiang opened this issue Sep 14, 2024 · 1 comment
Open

vit分类模型,与pt模型相比精度掉点太大 #1575

BaseJiang opened this issue Sep 14, 2024 · 1 comment
Labels
wontfix This will not be worked on

Comments

@BaseJiang
Copy link

对比测试v8分类模型和vit分类模型
a.同样的推断工程
b.同样的转换方式pt-->onnx-->trtexec 转换为engine
c.同样的测试集

结果:
v8cls分类模型
a.原pt模型与engine推断结果相差无几
b.相同工程,在两个不同推断平台测试的两个engine,结果几乎没有任何差距。

vit分类模型
a. 有些类别推断结果相差很大存在20%的差距,但是有些类别差距只有6%左右。
b.相同工程,两个不同推断平台下,两个engine,推断出的结果也存在不小差距。

问题:
请问这是模型特征表达有问题(但是原pt模型分类结果都还挺高),还是推断的数据前处理有问题?有需要的话,可以给您发vit onnx模型文件.

Copy link

stale bot commented Nov 27, 2024

This issue has been automatically marked as stale because it has not had recent activity. It will be closed if no further activity occurs. Thank you for your contributions.

@stale stale bot added the wontfix This will not be worked on label Nov 27, 2024
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
wontfix This will not be worked on
Projects
None yet
Development

No branches or pull requests

1 participant