Image Pre-processing Tools

[Samyssmile]

This is blocked by: #43

Add Image Pre-processing Tools for Augmentation and Resizing

Description

In the context of developing and optimizing Convolutional Neural Networks (CNNs) #43 it is essential to implement a suite of image pre-processing tools that can facilitate streamlined and effective model training. Specifically, we require utility classes/methods that address common tasks like image augmentation and resizing.

Image Augmentation enriches the dataset, allowing the model to be trained on more diverse data without collecting new images. Consequently, it helps improve the model's robustness and ability to generalize, thus reducing overfitting.

Requirements:

1. Image Augmentation: Implement tools or utilities that support various image augmentation techniques, such as:
   • Rotation
   • Flipping
   • Shearing
   • Zooming
   • Color variations (brightness, saturation, etc.)
2. Image Resizing: Provide functionality for resizing images, which should support:
   • Specifying target size
   • Maintaining the aspect ratio or optionally ignoring it
   • Different interpolation methods (e.g., nearest-neighbor, bilinear, etc.)

Acceptance Criteria:

• Implement augmentation tools that provide at least the above-mentioned functionalities.
• Implement resizing tools with the specified features.
• Ensure that the tools are adaptable and can be easily integrated into the existing workflow with CNNs.
• Provide documentation for each tool/utility, detailing the usage and potential application scenarios.
• Create unit tests to verify the functionality and reliability of the implemented tools.

Additional Context:

Ensuring robust and versatile image pre-processing tools can significantly enhance the model training phase, enabling the CNN to generalize better through exposure to varied and transformed data. This enhancement will bolster our ongoing work in developing CNN models for our project.

Tasks:

• Define the API for image augmentation and resizing tools.

• Develop the image augmentation tools.

• Develop the image resizing tools.

• Create documentation and usage examples.

• Implement unit tests.

• Conduct a review and testing phase to ensure functionality and integration capability.

• https://albumentations.ai/docs/introduction/image_augmentation/#:~:text=Image%20augmentation%20is%20a%20process,slightly%20change%20the%20original%20image.