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

updating:rotation augmentation is provided for machine based reading order

Browse source
This commit is contained in:
vahidrezanezhad 2025-04-17 00:12:30 +02:00
parent 77ef68a1ec
commit 4cc5aea166
1 changed files with 2 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

2
utils.py
View file

@ -455,6 +455,8 @@ def generate_arrays_from_folder_reading_order(classes_file_dir, modal_dir, batch
for thetha_i in thetha:
img_rot = rotation_not_90_func_single_image(img, thetha_i)
img_rot = resize_image(img_rot, height, width)
ret_x[batchcount, :,:,0] = img_rot[:,:,0]/3.0
ret_x[batchcount, :,:,2] = img_rot[:,:,2]/3.0
ret_x[batchcount, :,:,1] = img_rot[:,:,1]/5.0