search

DataXujing / ScaledYOLOv4

:fire::fire::fire: Scaled-YOLOv4训练自己的数据集详细教程PDF，关于paper解读请联系小编获取PDF文档
72 stars 24 forks source link

TensorRt使用自定数据集inference得到得结果不对 #7

Open caiduoduo12138 opened 3 years ago

caiduoduo12138 commented 3 years ago

你好，我在使用TensorRt输出结果时发现图片上的标注很密集，我去测试了test.py 可以得到正确结果，使用TensorRT时coco.names文件也修改了，可以帮我看看是什么原因吗，谢谢~

DataXujing commented 3 years ago

后处理是不是正确比如有没有做NMS? 请自行检查 前处理，后处理代码，很可能问题出在TensorRT output后的后处理部分。