From ed052a93dca8e237f5d4d7cc5eb6180fc6c4cb09 Mon Sep 17 00:00:00 2001
From: KRG <rgkannan676@gmail.com>
Date: Wed, 3 Jul 2024 09:57:24 +0800
Subject: [PATCH] Update yolov8_seg.cpp for 'get_downscale_rect' to fix empty
 mask issue. (#1549)

Fix issue for https://github.com/wang-xinyu/tensorrtx/issues/1547
---
 yolov8/yolov8_seg.cpp | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/yolov8/yolov8_seg.cpp b/yolov8/yolov8_seg.cpp
index 5a86f97..415d4e7 100644
--- a/yolov8/yolov8_seg.cpp
+++ b/yolov8/yolov8_seg.cpp
@@ -23,8 +23,8 @@ static cv::Rect get_downscale_rect(float bbox[4], float scale) {
 
     left = left < 0 ? 0 : left;
     top = top < 0 ? 0 : top;
-    right = right > 640 ? 640 : right;
-    bottom = bottom > 640 ? 640 : bottom;
+    right = right > kInputW ? kInputW : right;
+    bottom = bottom > kInputH ? kInputH : bottom;
 
     left /= scale;
     top /= scale;