tensorrtx icon indicating copy to clipboard operation
tensorrtx copied to clipboard

Published 20 hours ago • verified publisher icon wang-xinyu

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

Open BouncyKnight12 opened this issue 1 year ago • 5 comments

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

BouncyKnight12 avatar Jan 20 '24 09:01 BouncyKnight12

看看confidence是多少？

wang-xinyu avatar Jan 22 '24 03:01 wang-xinyu

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

BouncyKnight12 avatar Jan 22 '24 07:01 BouncyKnight12

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

BouncyKnight12 avatar Jan 26 '24 09:01 BouncyKnight12

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

wang-xinyu avatar Jan 30 '24 05:01 wang-xinyu

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[bot] avatar Apr 16 '24 05:04 stale[bot]