From 0c919d17dbb63196622bc2828293f9b3b71f26bb Mon Sep 17 00:00:00 2001
From: johnlockejrr <16368414+johnlockejrr@users.noreply.github.com>
Date: Wed, 14 May 2025 03:27:12 -0700
Subject: [PATCH] Update utils.py

Safely get filename without extension
---
 utils.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/utils.py b/utils.py
index 783f0d3..ed8b34a 100644
--- a/utils.py
+++ b/utils.py
@@ -158,7 +158,7 @@ def data_gen(img_folder, mask_folder, batch_size, input_height, input_width, n_c
             # print(img_folder+'/'+n[i])
 
             try:
-                filename = n[i].split('.')[0]
+                filename = os.path.splitext(n[i])[0]
 
                 train_img = cv2.imread(img_folder + '/' + n[i]) / 255.
                 train_img = cv2.resize(train_img, (input_width, input_height),