From 86b009bc319835fc6b4888b1f6c215a7d7b7a15b Mon Sep 17 00:00:00 2001 From: Robert Sachunsky Date: Tue, 24 Feb 2026 20:41:08 +0100 Subject: [PATCH] training.utils.preprocess_imgs: fix file name stemming 27f43c17 --- src/eynollah/training/utils.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/eynollah/training/utils.py b/src/eynollah/training/utils.py index f059354..818fd14 100644 --- a/src/eynollah/training/utils.py +++ b/src/eynollah/training/utils.py @@ -787,8 +787,8 @@ def preprocess_imgs(config, # labs_list not used because stem matching more robust indexer = 0 for img, lab in tqdm(zip(imgs_list, labs_list)): - img = cv2.imread(os.path.join(dir_img, img)) img_name = os.path.splitext(img)[0] + img = cv2.imread(os.path.join(dir_img, img)) if config['task'] in ["segmentation", "binarization"]: # assert lab == img_name + '.png' lab = cv2.imread(os.path.join(dir_lab, img_name + '.png'))