From 248e4a81b2bb9a1beaf2c5414831119d136bcbbf Mon Sep 17 00:00:00 2001
From: psychedelicious <4822129+psychedelicious@users.noreply.github.com>
Date: Wed, 11 Sep 2024 20:07:24 +1000
Subject: [PATCH] fix(nodes): handle no detected line segments

---
 invokeai/backend/image_util/mlsd/utils.py | 14 +++++++++-----
 1 file changed, 9 insertions(+), 5 deletions(-)

diff --git a/invokeai/backend/image_util/mlsd/utils.py b/invokeai/backend/image_util/mlsd/utils.py
index aa7f866c0b..dbe9a98d09 100644
--- a/invokeai/backend/image_util/mlsd/utils.py
+++ b/invokeai/backend/image_util/mlsd/utils.py
@@ -79,11 +79,15 @@ def pred_lines(image, model,
             y_end = y + disp_y_end
             segments_list.append([x_start, y_start, x_end, y_end])
 
-    lines = 2 * np.array(segments_list)  # 256 > 512
-    lines[:, 0] = lines[:, 0] * w_ratio
-    lines[:, 1] = lines[:, 1] * h_ratio
-    lines[:, 2] = lines[:, 2] * w_ratio
-    lines[:, 3] = lines[:, 3] * h_ratio
+    if segments_list:
+        lines = 2 * np.array(segments_list)  # 256 > 512
+        lines[:, 0] = lines[:, 0] * w_ratio
+        lines[:, 1] = lines[:, 1] * h_ratio
+        lines[:, 2] = lines[:, 2] * w_ratio
+        lines[:, 3] = lines[:, 3] * h_ratio
+    else:
+        # No segments detected - return empty array
+        lines = np.array([])
 
     return lines