Update process_img.py

process_img.py

@@ -14,7 +14,7 @@ class Image_Processor:
         white_pixel_percentage = (white_pixels_count / total_pixels) * 100
         return white_pixel_percentage > white_threshold
 
-    def padding_white(self,image, output_size=(224, 224)):
+    def padding_white(self,image, output_size=(336, 336)):
         # Ensure image is in RGB mode before padding
         if image.mode != 'RGB':
             image = image.convert('RGB')
@@ -41,7 +41,8 @@ class Image_Processor:
             final_image = self.padding_white(resized_image)
             logging.info('Resized and Padded Image')
         else:
-            final_image = self.resize_image_with_aspect_ratio(image)
+            #final_image = self.resize_image_with_aspect_ratio(image)
+            final_image = image.resize((336,336))
             logging.info('Resized Image')
         
         final_image = final_image.convert('L') if final_image.mode != 'L' else final_image