UNeedCryDear/yolov8-opencv-onnxruntime-cpp

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

Closed this issue · 1 comments

ZJDATY commented

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

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