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

Safely get filename without extension
---
 utils.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/utils.py b/utils.py
index 8be6963..b727207 100644
--- a/utils.py
+++ b/utils.py
@@ -434,7 +434,7 @@ def generate_arrays_from_folder_reading_order(classes_file_dir, modal_dir, batch
     batchcount = 0
     while True:
         for i in all_labels_files:
-            file_name = i.split('.')[0]
+            file_name = os.path.splitext(i)[0]
             img = cv2.imread(os.path.join(modal_dir,file_name+'.png'))
 
             label_class = int( np.load(os.path.join(classes_file_dir,i)) )
@@ -479,7 +479,7 @@ def data_gen(img_folder, mask_folder, batch_size, input_height, input_width, n_c
 
         for i in range(c, c + batch_size):  # initially from 0 to 16, c = 0.
             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),