mirror of
https://github.com/qurator-spk/sbb_pixelwise_segmentation.git
synced 2025-06-09 11:50:04 +02:00
updating:rotation augmentation is provided for machine based reading order
This commit is contained in:
parent
77ef68a1ec
commit
4cc5aea166
1 changed files with 2 additions and 0 deletions
2
utils.py
2
utils.py
|
@ -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
|
||||
|
|
Loading…
Add table
Add a link
Reference in a new issue