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

yolov8s在960x960输入size存在较大概率误检,同一个模型用onnx或者torch推理正常的 #1426

Open
asflower opened this issue Jan 2, 2024 · 6 comments

Comments

@asflower
Copy link

asflower commented Jan 2, 2024

Env

  • GPU, e.g. V100, RTX2080, TX2, Xavier NX, Nano, etc.
    GTX1660S、RTX2080Ti
  • OS, e.g. Ubuntu16.04, Win10, etc.
    Ubuntu1804
  • Cuda version
    cuda-11.4
  • TensorRT version
    8.5

About this repo

  • which branch/tag/commit are you using?
    master
  • which model? yolov5, retinaface?
    yolov8s

Your problem

1、yolov8s,960x960的模型,当输入图像小于960x960时,较大概率一些类别存在误检情况;
2、同一个模型,改成640x640输入,误检消失;
3、同一个模型、同一批图像,用onnx或者torch推理正常,没有误检;

@wang-xinyu
Copy link
Owner

@lindsayshuo 帮忙确认下呢

@lindsayshuo
Copy link
Contributor

训练的参数我看看

@asflower
Copy link
Author

不好意思,我这边对模型训练不是太了解,这里只做工程部署。
我怀疑问题可能出在编译的engine模型上,之前也尝试过替换前处理,问题还是存在

Copy link

stale bot commented Mar 13, 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 Mar 13, 2024
@RaidenCJ
Copy link

你好,请问找到原因了么? 方便共享下解决方案么 @asflower 我也遇到了这个问题

@stale stale bot removed the wontfix This will not be worked on label Apr 18, 2024
@lindsayshuo
Copy link
Contributor

已解决,看最新代码

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

No branches or pull requests

4 participants