Update utils.py

Safely get filename without extension
1 week ago · 0c919d17db
parent 479a6705c1
commit 0c919d17db
1 changed files with 1 additions and 1 deletions
--- 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),