qurator-spk/eynollah
1
0
Fork 0
mirror of https://github.com/qurator-spk/eynollah.git synced 2026-03-02 05:11:57 +01:00
Code Issues Projects Releases Wiki Activity

training.utils.preprocess_imgs: fix file name stemming 27f43c17

Browse source
This commit is contained in:
Robert Sachunsky 2026-02-24 20:41:08 +01:00
parent 20a3672be3
commit 86b009bc31
1 changed files with 1 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

2
src/eynollah/training/utils.py
View file

@ -787,8 +787,8 @@ def preprocess_imgs(config,
# labs_list not used because stem matching more robust # labs_list not used because stem matching more robust
indexer = 0 indexer = 0
for img, lab in tqdm(zip(imgs_list, labs_list)): 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_name = os.path.splitext(img)[0]
img = cv2.imread(os.path.join(dir_img, img))
if config['task'] in ["segmentation", "binarization"]: if config['task'] in ["segmentation", "binarization"]:
# assert lab == img_name + '.png' # assert lab == img_name + '.png'
lab = cv2.imread(os.path.join(dir_lab, img_name + '.png')) lab = cv2.imread(os.path.join(dir_lab, img_name + '.png'))