UNeedCryDear / yolov8-opencv-onnxruntime-cpp

yolov8 hub,cpp with onnxruntime and opencv
Apache License 2.0
319 stars 56 forks source link

大佬 我在导出yolov8-obb 模型时,添加了 nms=true 。 我想知道推理代码要怎么改, 直接去掉NMS函数吗? #42

Closed ZJDATY closed 2 months ago

ZJDATY commented 8 months ago

代码里能否检测模型启用了NMS,从而进行判断,是否走nms的流程。

UNeedCryDear commented 8 months ago

可以直接去掉nms的部分,直接出来的box和classid,conf就是最终的结果了,你可以对比下这两个结果有什么不一样的地方。