fix the condition if masks are empty

csrc/jetson/segment/include/yolov8-seg.hpp

@@ -428,36 +428,41 @@ void YOLOv8_seg::postprocess(std::vector<Object>& objs,
 		objs.push_back(obj);
 		cnt += 1;
 	}
+	if(masks.empty())
+	{
+		//masks is empty
+	}
+	else
+	{
+		cv::Mat matmulRes = (masks * protos).t();
+		cv::Mat maskMat = matmulRes.reshape(indices.size(), { seg_w, seg_h });
 
-	cv::Mat matmulRes = (masks * protos).t();
-	cv::Mat maskMat = matmulRes.reshape(indices.size(), { seg_w, seg_h });
-
-	std::vector<cv::Mat> maskChannels;
-	cv::split(maskMat, maskChannels);
-	int scale_dw = dw / input_w * seg_w;
-	int scale_dh = dh / input_h * seg_h;
+		std::vector<cv::Mat> maskChannels;
+		cv::split(maskMat, maskChannels);
+		int scale_dw = dw / input_w * seg_w;
+		int scale_dh = dh / input_h * seg_h;
 
-	cv::Rect roi(
-		scale_dw,
-		scale_dh,
-		seg_w - 2 * scale_dw,
-		seg_h - 2 * scale_dh);
+		cv::Rect roi(
+			scale_dw,
+			scale_dh,
+			seg_w - 2 * scale_dw,
+			seg_h - 2 * scale_dh);
 
-	for (int i = 0; i < indices.size(); i++)
-	{
-		cv::Mat dest, mask;
-		cv::exp(-maskChannels[i], dest);
-		dest = 1.0 / (1.0 + dest);
-		dest = dest(roi);
-		cv::resize(
-			dest,
-			mask,
-			cv::Size((int)width, (int)height),
-			cv::INTER_LINEAR
-		);
-		objs[i].boxMask = mask(objs[i].rect) > 0.5f;
+		for (int i = 0; i < indices.size(); i++)
+		{
+			cv::Mat dest, mask;
+			cv::exp(-maskChannels[i], dest);
+			dest = 1.0 / (1.0 + dest);
+			dest = dest(roi);
+			cv::resize(
+				dest,
+				mask,
+				cv::Size((int)width, (int)height),
+				cv::INTER_LINEAR
+			);
+			objs[i].boxMask = mask(objs[i].rect) > 0.5f;
+		}
 	}
-
 }
 
 void YOLOv8_seg::draw_objects(const cv::Mat& image,

csrc/segment/normal/include/yolov8-seg.hpp

@@ -459,33 +459,40 @@ void YOLOv8_seg::postprocess(std::vector<Object>& objs,
 		cnt += 1;
 	}
 
-	cv::Mat matmulRes = (masks * protos).t();
-	cv::Mat maskMat = matmulRes.reshape(indices.size(), { seg_w, seg_h });
+	if(masks.empty())
+	{
+		//masks is empty
+	}
+	else
+	{
+		cv::Mat matmulRes = (masks * protos).t();
+		cv::Mat maskMat = matmulRes.reshape(indices.size(), { seg_w, seg_h });
 
-	std::vector<cv::Mat> maskChannels;
-	cv::split(maskMat, maskChannels);
-	int scale_dw = dw / input_w * seg_w;
-	int scale_dh = dh / input_h * seg_h;
+		std::vector<cv::Mat> maskChannels;
+		cv::split(maskMat, maskChannels);
+		int scale_dw = dw / input_w * seg_w;
+		int scale_dh = dh / input_h * seg_h;
 
-	cv::Rect roi(
-		scale_dw,
-		scale_dh,
-		seg_w - 2 * scale_dw,
-		seg_h - 2 * scale_dh);
+		cv::Rect roi(
+			scale_dw,
+			scale_dh,
+			seg_w - 2 * scale_dw,
+			seg_h - 2 * scale_dh);
 
-	for (int i = 0; i < indices.size(); i++)
-	{
-		cv::Mat dest, mask;
-		cv::exp(-maskChannels[i], dest);
-		dest = 1.0 / (1.0 + dest);
-		dest = dest(roi);
-		cv::resize(
-			dest,
-			mask,
-			cv::Size((int)width, (int)height),
-			cv::INTER_LINEAR
-		);
-		objs[i].boxMask = mask(objs[i].rect) > 0.5f;
+		for (int i = 0; i < indices.size(); i++)
+		{
+			cv::Mat dest, mask;
+			cv::exp(-maskChannels[i], dest);
+			dest = 1.0 / (1.0 + dest);
+			dest = dest(roi);
+			cv::resize(
+				dest,
+				mask,
+				cv::Size((int)width, (int)height),
+				cv::INTER_LINEAR
+			);
+			objs[i].boxMask = mask(objs[i].rect) > 0.5f;
+		}
 	}
 
 }

csrc/segment/simple/include/yolov8-seg.hpp

@@ -431,35 +431,41 @@ void YOLOv8_seg::postprocess(std::vector<Object>& objs,
 		cnt += 1;
 	}
 
-	cv::Mat matmulRes = (masks * protos).t();
-	cv::Mat maskMat = matmulRes.reshape(indices.size(), { seg_w, seg_h });
+	if(masks.empty())
+	{
+		//masks is empty
+	}
+	else
+	{
+		cv::Mat matmulRes = (masks * protos).t();
+		cv::Mat maskMat = matmulRes.reshape(indices.size(), { seg_w, seg_h });
 
-	std::vector<cv::Mat> maskChannels;
-	cv::split(maskMat, maskChannels);
-	int scale_dw = dw / input_w * seg_w;
-	int scale_dh = dh / input_h * seg_h;
+		std::vector<cv::Mat> maskChannels;
+		cv::split(maskMat, maskChannels);
+		int scale_dw = dw / input_w * seg_w;
+		int scale_dh = dh / input_h * seg_h;
 
-	cv::Rect roi(
-		scale_dw,
-		scale_dh,
-		seg_w - 2 * scale_dw,
-		seg_h - 2 * scale_dh);
+		cv::Rect roi(
+			scale_dw,
+			scale_dh,
+			seg_w - 2 * scale_dw,
+			seg_h - 2 * scale_dh);
 
-	for (int i = 0; i < indices.size(); i++)
-	{
-		cv::Mat dest, mask;
-		cv::exp(-maskChannels[i], dest);
-		dest = 1.0 / (1.0 + dest);
-		dest = dest(roi);
-		cv::resize(
-			dest,
-			mask,
-			cv::Size((int)width, (int)height),
-			cv::INTER_LINEAR
-		);
-		objs[i].boxMask = mask(objs[i].rect) > 0.5f;
+		for (int i = 0; i < indices.size(); i++)
+		{
+			cv::Mat dest, mask;
+			cv::exp(-maskChannels[i], dest);
+			dest = 1.0 / (1.0 + dest);
+			dest = dest(roi);
+			cv::resize(
+				dest,
+				mask,
+				cv::Size((int)width, (int)height),
+				cv::INTER_LINEAR
+			);
+			objs[i].boxMask = mask(objs[i].rect) > 0.5f;
+		}
 	}
-
 }
 
 void YOLOv8_seg::draw_objects(const cv::Mat& image,