From be57f137d77946c5ed81cb001e23964d6bf27d9e Mon Sep 17 00:00:00 2001
From: johnlockejrr <16368414+johnlockejrr@users.noreply.github.com>
Date: Sun, 11 May 2025 05:31:34 -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 3d42b64..cba20c2 100644
--- a/utils.py
+++ b/utils.py
@@ -667,7 +667,7 @@ def provide_patches(imgs_list_train, segs_list_train, dir_img, dir_seg, dir_flow
indexer = 0
for im, seg_i in tqdm(zip(imgs_list_train, segs_list_train)):
- img_name = im.split('.')[0]
+ img_name = os.path.splitext(im)[0]
if task == "segmentation" or task == "binarization":
dir_of_label_file = os.path.join(dir_seg, img_name + '.png')
elif task=="enhancement":