search

FeiYull / TensorRT-Alpha

🔥🔥🔥TensorRT for YOLOv8、YOLOv8-Pose、YOLOv8-Seg、YOLOv8-Cls、YOLOv7、YOLOv6、YOLOv5、YOLONAS......🚀🚀🚀CUDA IS ALL YOU NEED.🍎🍎🍎
GNU General Public License v2.0
1.26k stars 196 forks source link

yolov6模型转换正常，可以正常推理，但是没有目标检测框出来，比较奇怪，找不到原因 #37

Closed chennankuan closed 1 year ago

chennankuan commented 1 year ago

遇到这样一种情况：yolov6模型转换后推理，但是没有检测框出来，全部检测结果为无目标。 另外试了下yolov6n.onnx转换到trt，发现也是一样的没有目标框。 本来是以为是tensorrt和cuda/cudnn的版本有冲突引起的，但是发现没有冲突的情况下，也是一样的现象。

chennankuan commented 1 year ago

if (!setInputstream(source, image path, video path, camera id,capture, total batches, delay, time, param)) ，原来是注释了这个函数，导致param.src h = capture.get(cv::CAP PROP FRAME HEIGHT):param.src w = capture.get(cv::CAP PROP FRAME WIDTH); src_h和src_w没有获取到数据，这样模型能正常推理，但是就是没有结果。