mirror of
https://github.com/qurator-spk/sbb_pixelwise_segmentation.git
synced 2025-07-01 22:50:15 +02:00
Update utils.py
This commit is contained in:
parent
02b1436f39
commit
479a6705c1
1 changed files with 1 additions and 1 deletions
2
utils.py
2
utils.py
|
@ -378,7 +378,7 @@ def provide_patches(dir_img, dir_seg, dir_flow_train_imgs,
|
|||
|
||||
indexer = 0
|
||||
for im, seg_i in tqdm(zip(imgs_cv_train, segs_cv_train)):
|
||||
img_name = im.split('.')[0]
|
||||
img_name = os.path.splitext(im)[0]
|
||||
if not patches:
|
||||
cv2.imwrite(dir_flow_train_imgs + '/img_' + str(indexer) + '.png',
|
||||
resize_image(cv2.imread(dir_img + '/' + im), input_height, input_width))
|
||||
|
|
Loading…
Add table
Add a link
Reference in a new issue