update augment + YOLO pipeline

src/utils.py

@@ -41,14 +41,11 @@ def load_image(image_path: Path, target_size: Tuple[int, int] = None) -> Optiona
         image = cv2.imread(str(image_path))
         if image is None:
             return None
-        
         # Convert BGR to RGB
         image = cv2.cvtColor(image, cv2.COLOR_BGR2RGB)
-        
         # Resize if target_size is provided
         if target_size:
             image = cv2.resize(image, target_size, interpolation=cv2.INTER_AREA)
-        
         return image
     except Exception as e:
         print(f"Error loading image {image_path}: {e}")