From 479a6705c1d4c3892d8170f48fd249bb60da2044 Mon Sep 17 00:00:00 2001
From: johnlockejrr <16368414+johnlockejrr@users.noreply.github.com>
Date: Sat, 10 May 2025 14:58:40 -0700
Subject: [PATCH] Update utils.py

---
 utils.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/utils.py b/utils.py
index 7c65f18..783f0d3 100644
--- a/utils.py
+++ b/utils.py
@@ -378,7 +378,7 @@ def provide_patches(dir_img, dir_seg, dir_flow_train_imgs,
 
     indexer = 0
     for im, seg_i in tqdm(zip(imgs_cv_train, segs_cv_train)):
-        img_name = im.split('.')[0]
+        img_name = os.path.splitext(im)[0]
         if not patches:
             cv2.imwrite(dir_flow_train_imgs + '/img_' + str(indexer) + '.png',
                         resize_image(cv2.imread(dir_img + '/' + im), input_height, input_width))