Monday-Leo
Monday-Leo
因为tensorrt 8.2有bug。升级至8.4可以解决。
这是nms的bug，yolov5的nms没有并入到模型当中，所以没关系。
那就没有问题，甚至可以用tensorrt7，这个仓库主要是添加了efficientnms到模型中，所以才使用的tensorrt8.4。
linux下删除include文件夹，dirent.h是自带的，并且使用cmakelist_linux.txt
降低pytroch版本至1.7.1再导出ONNX模型。torch1.10-cu102会出这个问题。
The coco dataset is so large that I cannot download. So sorry, you can try it by yourself.
Do you keep the IOU and score threshold of NMS the same? The defaults are 0.45 and 0.25. This affects the Map. 3% is a relatively large accuracy gap.
不需要，用cpu导出onnx即可，提供一下完整的报错代码。
这个报错我暂时还没有遇到过，是官方的模型，还是自己训练的模型？能否用netron可视化一下输入输出。
你把转换好的onnx发我，我试试转话一下吧，定位一下问题。