qurator-spk/sbb_pixelwise_segmentation
1
0
Fork 0
mirror of https://github.com/qurator-spk/sbb_pixelwise_segmentation.git synced 2025-06-09 20:00:05 +02:00
Code Issues Releases Wiki Activity

brightness augmentation modified

Browse source
This commit is contained in:
b-vr103 2024-07-17 18:20:24 +02:00
parent 9521768774
commit f2692cf8dd
1 changed files with 8 additions and 5 deletions
Download patch file Download diff file Expand all files Collapse all files

3
utils.py
View file

@ -599,12 +599,15 @@ def provide_patches(imgs_list_train, segs_list_train, dir_img, dir_seg, dir_flow
indexer += 1 indexer += 1
if brightening: if brightening:
for factor in brightness: for factor in brightness:
try:
cv2.imwrite(dir_flow_train_imgs + '/img_' + str(indexer) + '.png', cv2.imwrite(dir_flow_train_imgs + '/img_' + str(indexer) + '.png',
(resize_image(do_brightening(dir_img + '/' +im, factor), input_height, input_width))) (resize_image(do_brightening(dir_img + '/' +im, factor), 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))
indexer += 1 indexer += 1
except:
pass
if binarization: if binarization:
cv2.imwrite(dir_flow_train_imgs + '/img_' + str(indexer) + '.png', cv2.imwrite(dir_flow_train_imgs + '/img_' + str(indexer) + '.png',