File tree Expand file tree Collapse file tree 1 file changed +6
-0
lines changed
Expand file tree Collapse file tree 1 file changed +6
-0
lines changed Original file line number Diff line number Diff line change @@ -132,6 +132,8 @@ void Utils::PostProcess(std::vector<Ort::Value>& output_tensors, const cv::Mat&
132132 if (output_tensors.empty ())
133133 {
134134 CONSOLE_BRIDGE_logError (" [SAM]: Decoder returned no outputs." );
135+ // Push empty mask to preserve index alignment with boxes (one mask slot per box).
136+ result.masks .push_back (cv::Mat ());
135137 return ;
136138 }
137139
@@ -148,6 +150,8 @@ void Utils::PostProcess(std::vector<Ort::Value>& output_tensors, const cv::Mat&
148150 if (masksIdx < 0 )
149151 {
150152 CONSOLE_BRIDGE_logError (" [SAM]: No 4D mask tensor found in decoder outputs." );
153+ // Push empty mask to preserve index alignment with boxes (one mask slot per box).
154+ result.masks .push_back (cv::Mat ());
151155 return ;
152156 }
153157
@@ -244,5 +248,7 @@ void Utils::PostProcess(std::vector<Ort::Value>& output_tensors, const cv::Mat&
244248 else
245249 {
246250 CONSOLE_BRIDGE_logError (" [SAM]: Unexpected mask tensor shape." );
251+ // Push empty mask to preserve index alignment with boxes (one mask slot per box).
252+ result.masks .push_back (cv::Mat ());
247253 }
248254}
You can’t perform that action at this time.
0 commit comments