From 4635dd219d5cfade1c038a371dceb78452a7fbf9 Mon Sep 17 00:00:00 2001 From: vahidrezanezhad Date: Thu, 17 Apr 2025 00:12:30 +0200 Subject: [PATCH] updating:rotation augmentation is provided for machine based reading order --- train/utils.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/train/utils.py b/train/utils.py index 485056b..8be6963 100644 --- a/train/utils.py +++ b/train/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