qurator-spk/eynollah
1
0
Fork 0
mirror of https://github.com/qurator-spk/eynollah.git synced 2025-10-07 06:59:58 +02:00
Code Issues Projects Releases Wiki Activity

using prepared binarized images in the case of augmentation

Browse source
This commit is contained in:
vahidrezanezhad 2024-08-22 21:58:09 +02:00
parent aeb2ee4e3e
commit 61cdd2acb8
1 changed files with 9 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

7
train/utils.py
View file

@ -690,6 +690,13 @@ def provide_patches(imgs_list_train, segs_list_train, dir_img, dir_seg, dir_flow
pass pass
if binarization: if binarization:
if dir_img_bin:
img_bin_corr = cv2.imread(dir_img_bin + '/' + img_name+'.png')
cv2.imwrite(dir_flow_train_imgs + '/img_' + str(indexer) + '.png',
resize_image(img_bin_corr, input_height, input_width))
else:
cv2.imwrite(dir_flow_train_imgs + '/img_' + str(indexer) + '.png', cv2.imwrite(dir_flow_train_imgs + '/img_' + str(indexer) + '.png',
resize_image(otsu_copy(cv2.imread(dir_img + '/' + im)), input_height, input_width)) resize_image(otsu_copy(cv2.imread(dir_img + '/' + im)), input_height, input_width))