Merge pull request #160 from Flippchen/main

1 year ago · f36a1fdaef
parent c1c7dadaf0 4bf9dd15ed
commit f36a1fdaef
2 changed files with 2 additions and 2 deletions
--- a/csrc/segment/normal/include/yolov8-seg.hpp
+++ b/csrc/segment/normal/include/yolov8-seg.hpp
@ -326,7 +326,7 @@ void YOLOv8_seg::postprocess(
    }
    else {
        cv::Mat matmulRes = (masks * protos).t();
-        cv::Mat maskMat   = matmulRes.reshape(indices.size(), {seg_w, seg_h});
+        cv::Mat maskMat   = matmulRes.reshape(indices.size(), {seg_h, seg_w});
        std::vector<cv::Mat> maskChannels;
        cv::split(maskMat, maskChannels);
--- a/csrc/segment/simple/include/yolov8-seg.hpp
+++ b/csrc/segment/simple/include/yolov8-seg.hpp
@ -300,7 +300,7 @@ void YOLOv8_seg::postprocess(
    }
    else {
        cv::Mat matmulRes = (masks * protos).t();
-        cv::Mat maskMat   = matmulRes.reshape(indices.size(), {seg_w, seg_h});
+        cv::Mat maskMat   = matmulRes.reshape(indices.size(), {seg_h, seg_w});
        std::vector<cv::Mat> maskChannels;
        cv::split(maskMat, maskChannels);