diff --git a/yolov8_seg_onnx.cpp b/yolov8_seg_onnx.cpp index 554fbd0..b268565 100644 --- a/yolov8_seg_onnx.cpp +++ b/yolov8_seg_onnx.cpp @@ -163,6 +163,7 @@ bool Yolov8SegOnnx::ReadModel(const std::string& modelPath, bool isCuda, int cud catch (const std::exception&) { return false; } + return true; } @@ -240,9 +241,9 @@ bool Yolov8SegOnnx::OnnxBatchDetect(std::vector& srcImgs, std::vector class_ids;//结果id数组 - std::vector confidences;//结果每个id对应置信度数组 - std::vector boxes;//每个id矩形框 + std::vector class_ids;//缁撴灉id鏁扮粍 + std::vector confidences;//缁撴灉姣忎釜id瀵瑰簲缃俊搴︽暟缁 + std::vector boxes;//姣忎釜id鐭╁舰妗 std::vector> picked_proposals; //output0[:,:, 5 + _className.size():net_width]===> for mask for (int r = 0; r < rows; ++r) { //stride cv::Mat scores(1, _className.size(), CV_32F, pdata + 4); @@ -264,7 +265,7 @@ bool Yolov8SegOnnx::OnnxBatchDetect(std::vector& srcImgs, std::vector nms_result; @@ -294,7 +295,7 @@ bool Yolov8SegOnnx::OnnxBatchDetect(std::vector& srcImgs, std::vector& srcImgs, std::vector