mirror of
https://github.com/qurator-spk/sbb_pixelwise_segmentation.git
synced 2025-06-09 11:50:04 +02:00
Update utils.py
This commit is contained in:
parent
451188c3b9
commit
be57f137d7
1 changed files with 1 additions and 1 deletions
2
utils.py
2
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":
|
||||
|
|
Loading…
Add table
Add a link
Reference in a new issue