YOLOv7_Tensorrt
YOLOv7_Tensorrt copied to clipboard
A simple implementation of Tensorrt YOLOv7
能否出一版C++的推理呢
我按照您的readme在开发板上进行了模型部署,一切顺利,但是检测物体的置信度为负数,这是因为什么原因造成的,因为我没有在trt推理下评估模型的指标,所以我很难确定这样带来的性能影响,如果你可以回答我的疑问,我将很感激
`D:\Anaconda3\envs\torch17\python.exe D:/work_place/yolov7_coil_test/infer_python.py [07/18/2022-18:04:34] [TRT] [I] [MemUsageChange] Init CUDA: CPU +578, GPU +0, now: CPU 8750, GPU 961 (MiB) [07/18/2022-18:04:34] [TRT] [I] Loaded engine size: 75 MiB [07/18/2022-18:04:34] [TRT] [W] TensorRT...
can you test the mAP of Yolov7 in coco2017val with TensorRT? Thank you.
无法生成 engine
使用export_onnx.py 文件生成了onnx文件,在使用trtexec 生成engine 文件时失败,环境为 cuda10.2 cudnn8.4.1 torch1.7.1 是否在生成onnx文件时需要将device 设置为GPU模式
[07/10/2022-18:30:00] [E] [TRT] ModelImporter.cpp:774: --- Begin node --- [07/10/2022-18:30:00] [E] [TRT] ModelImporter.cpp:775: input: "642" input: "1341" output: "645" name: "Mul_446" op_type: "Mul" [07/10/2022-18:30:00] [E] [TRT] ModelImporter.cpp:776: --- End node ---...
相关环境: python 3.6.9 cuda-10.2 cuDNN-8.4.1 TensorRT-8.4.3.1 转的是官方的yolov7.pt,报错如下,搜了下有人说是cuda-10.2的问题,需要装2个补丁,我装过了还是报这个错,求助下 ``` [10/12/2022-11:13:58] [W] [TRT] onnx2trt_utils.cpp:369: Your ONNX model has been generated with INT64 weights, while TensorRT does not natively support INT64. Attempting to...
运行infer.py后出现KeyError:'num_dets',请问应该怎么解决