mirror of
https://github.com/qurator-spk/sbb_pixelwise_segmentation.git
synced 2025-06-09 20:00:05 +02:00
using prepared binarized images in the case of augmentation
This commit is contained in:
parent
f31219b1c9
commit
9904846776
1 changed files with 9 additions and 2 deletions
11
utils.py
11
utils.py
|
@ -690,8 +690,15 @@ def provide_patches(imgs_list_train, segs_list_train, dir_img, dir_seg, dir_flow
|
||||||
pass
|
pass
|
||||||
|
|
||||||
if binarization:
|
if binarization:
|
||||||
cv2.imwrite(dir_flow_train_imgs + '/img_' + str(indexer) + '.png',
|
|
||||||
resize_image(otsu_copy(cv2.imread(dir_img + '/' + im)), input_height, input_width))
|
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',
|
||||||
|
resize_image(otsu_copy(cv2.imread(dir_img + '/' + im)), input_height, input_width))
|
||||||
|
|
||||||
cv2.imwrite(dir_flow_train_labels + '/img_' + str(indexer) + '.png',
|
cv2.imwrite(dir_flow_train_labels + '/img_' + str(indexer) + '.png',
|
||||||
resize_image(cv2.imread(dir_of_label_file), input_height, input_width))
|
resize_image(cv2.imread(dir_of_label_file), input_height, input_width))
|
||||||
|
|
Loading…
Add table
Add a link
Reference in a new issue