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

yolov8模型在1280*1280下出现多检 #1431

Closed
BouncyKnight12 opened this issue Jan 20, 2024 · 5 comments
Closed

yolov8模型在1280*1280下出现多检 #1431

BouncyKnight12 opened this issue Jan 20, 2024 · 5 comments
Labels
wontfix This will not be worked on

Comments

@BouncyKnight12
Copy link

Env

  • GPU, jetson-agx
  • OS, e.g. Ubuntu20.04
  • Cuda 11.4
  • TensorRT 8.5.2

About this repo

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

Your problem

使用yolov8n模型在1280*1280分辨率下进行推理,推理结果比pytorch的输出结果多了一个框
pytorch的结果如下
20240109104200_456_2_131
tensorrtx的结果如下
_20240109104200_456_2_131

@wang-xinyu
Copy link
Owner

看看confidence是多少?

@BouncyKnight12
Copy link
Author

在tensorrtx中的confidence是0.566019,使用pytorch设置的confidence是0.25并且打开了half开关

@BouncyKnight12
Copy link
Author

我使用tensorrt的命令行工具trtexec之间去导出模型也依然存在这样的问题
20240109104200_456_2_131_trtexec

@wang-xinyu
Copy link
Owner

有轻微diff是正常的,可以多跑一些,对比下mAP

Copy link

stale bot commented Apr 16, 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 Apr 16, 2024
@stale stale bot closed this as completed Jun 9, 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

2 participants