Merge pull request #446 from Nuzhny007/master

Fix yolov8 instance segmentation
Smorodov · Feb 15, 2024 · 1699dcf · 1699dcf
2 parents d4e8096 + 64bee78
commit 1699dcf
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 1 deletion.
diff --git a/src/Detector/tensorrt_yolo/YoloONNX.cpp b/src/Detector/tensorrt_yolo/YoloONNX.cpp
@@ -654,6 +654,9 @@ void YoloONNX::ProcessMaskOutput(size_t imgIdx, const std::vector<float*>& outpu
 
                 std::vector<float> temp_proto(output + k + 5 + nc, output + k + net_width);
                 picked_proposals.push_back(temp_proto);
+
+                if (rectBoxes.size() >= CV_CN_MAX - 1)
+                    break;
             }
         }
 

diff --git a/src/Tracker/track.cpp b/src/Tracker/track.cpp
@@ -1149,7 +1149,7 @@ void CTrack::CreateExternalTracker(int channels)
 #ifdef USE_OCV_KCF
         if (!m_tracker || m_tracker.empty())
         {
-#if (((CV_VERSION_MAJOR == 4) && (CV_VERSION_MINOR > 5)) || ((CV_VERSION_MAJOR == 4) && (CV_VERSION_MINOR == 5) && (CV_VERSION_REVISION > 2))  || (CV_VERSION_MAJOR > 4))
+#if (((CV_VERSION_MAJOR == 4) && (CV_VERSION_MINOR > 6)) || (CV_VERSION_MAJOR > 4))
             cv::TrackerNano::Params params;
             params.backbone = "nanotrack_backbone_sim.onnx";
             params.neckhead = "nanotrack_head_sim.onnx";