This project started on 26/09/2023
and was completed within 2 weeks on 05/10/2023
for AItaca Tech-Test-ML-Hand.
Primary Objective:
- Develop a segmentation model that removes hands from the background with high precision.
Secondary Objectives:
- Implement a web-based application using Streamlit, using user-input hand images and segmented images as the output.
- Data Cleaning:
- Created two folders named 'original' and 'no_bg'. Each hand image in these folders has a unique ID which is the original folder name appended with its index ([:10]).
- Examined image sizes and ratios. Chose ratios [1.87, 1.85] to resize images to dimensions (2160, 4000) for the model.
- Manually reviewed 'no_bg' images, discarding those that did not display accurate masks (like other body parts or the background).
- The final dataset consisted of 79 unique original and 'no_bg' images. 'No_bg' images were transformed into true masks for predictions.
- Data Split:
- Initially, the dataset was split in an 80/10/10 ratio for training, validation, and testing respectively. Used Data Augmentation to produce 3 unique new images for training.
- For subsequent models post the MVP, adopted a 65/15/10 split. Enhanced Data Augmentation to generate 5 unique new images for training and allocated more images for validation.
- Data Augmentation:
Parameters used:
- Brightness Range: 0.5, 1.5
- Zoom Range: 0.2
- Rotation Range: 30
- Width & Height shift Range: 0.1
- Horizontal Flip: True
- Shear Range: 0.1
- Image Pre-Processing:
Tried multiple pre-processing techniques, unprocessed images were eventually used as they yielded better performance.
- Normalization: Pixel Normalization
- Noise reduction: Bilateral Filtering
- Histogram Equalization: CLAHE
- Edge detection: Canny Edges
- Morphological Operations: Dilatation
![Screenshot 2023-10-04 at 19 54 09](https://private-user-images.githubusercontent.com/90038586/272667077-7dd521cd-d455-4b44-bbeb-e05c37ea60ed.png?jwt=eyJhbGciOiJIUzI1NiIsInR5cCI6IkpXVCJ9.eyJpc3MiOiJnaXRodWIuY29tIiwiYXVkIjoicmF3LmdpdGh1YnVzZXJjb250ZW50LmNvbSIsImtleSI6ImtleTUiLCJleHAiOjE3MjQwNTg3MzMsIm5iZiI6MTcyNDA1ODQzMywicGF0aCI6Ii85MDAzODU4Ni8yNzI2NjcwNzctN2RkNTIxY2QtZDQ1NS00YjQ0LWJiZWItZTA1YzM3ZWE2MGVkLnBuZz9YLUFtei1BbGdvcml0aG09QVdTNC1ITUFDLVNIQTI1NiZYLUFtei1DcmVkZW50aWFsPUFLSUFWQ09EWUxTQTUzUFFLNFpBJTJGMjAyNDA4MTklMkZ1cy1lYXN0LTElMkZzMyUyRmF3czRfcmVxdWVzdCZYLUFtei1EYXRlPTIwMjQwODE5VDA5MDcxM1omWC1BbXotRXhwaXJlcz0zMDAmWC1BbXotU2lnbmF0dXJlPThhNGQ5NGJiY2U0NjVkMjk2NGRmNDI1NzNjZWY1OGI4MjYzZDczNTc0YWQyYzMwOGU2NTM0YjJlNGZhYWI1ZjgmWC1BbXotU2lnbmVkSGVhZGVycz1ob3N0JmFjdG9yX2lkPTAma2V5X2lkPTAmcmVwb19pZD0wIn0.742LuCBVX71OKAhi08Rx9qtWd6O-GLYAL45_-H85H38)
- Minimum Viable Product: MPV
- New Prototype (Explained all the Steps and cleaned MPV code): Prototype
- Final model Eva X
- The MPV was done and finished in the first week, on
29/09/2023
, showing some promising results. - Pre-processing the images did not improve performance.
- Succeeding models were architecturally more intricate, with a limit of 128 filters for optimum performance. Incorporating dropout and residual layers was beneficial for generating prediction masks.
- While there was an intent to experiment with various transfer learning models, time constraints prevented this. Another potential enhancement considered was using object detection before segmentation and fine-tuning some hyperparameters.
- JupyterLab: Enviorment for Python scripts and managing files.
Libraries
📚 Basic Libraries
- Numpy: Image numeric array manipulation
- Os: File access.
- Matplotlib: Visualization.
- Shutil: Folder operations (copying, deleting...).
- Random: To generate random subsets of data.
- Warnings: Roses are red, violets are blue --> Warnings are annoying.
🌐 Computer Vision
- TensorFlow: Machine Learning for Computer Vision.
- Keras: High-level neural networks API for Deep Learning, running on top of TensorFlow.
- ImageDataGenerator: To generate random data augmentation (flips, zoom...).
The architecture I've implemented is a variant of the U-Net model, a popular architecture for semantic image segmentation:
# First Downsampling Block x = layers.Conv2D(32, (3, 3), activation='relu', padding='same')(inputs) x = layers.BatchNormalization()(x) x = layers.Conv2D(32, (3, 3), activation='relu', padding='same')(x) x = layers.BatchNormalization()(x) x = layers.Conv2D(32, (3, 3), activation='relu', padding='same')(x) x = layers.BatchNormalization()(x) x = layers.Dropout(0.2)(x) residual_1 = x x = layers.MaxPooling2D((2, 2))(x) # Second Downsampling Block x = layers.Conv2D(64, (3, 3), activation='relu', padding='same')(x) x = layers.BatchNormalization()(x) x = layers.Conv2D(64, (3, 3), activation='relu', padding='same')(x) x = layers.BatchNormalization()(x) x = layers.Conv2D(64, (3, 3), activation='relu', padding='same')(x) x = layers.BatchNormalization()(x) x = layers.Dropout(0.3)(x) residual_2 = x x = layers.MaxPooling2D((2, 2))(x) # Bottleneck x = layers.Conv2D(128, (3, 3), activation='relu', padding='same')(x) x = layers.BatchNormalization()(x) x = layers.Conv2D(128, (3, 3), activation='relu', padding='same')(x) x = layers.BatchNormalization()(x) x = layers.Conv2D(128, (3, 3), activation='relu', padding='same')(x) x = layers.BatchNormalization()(x) x = layers.Dropout(0.4)(x) # Upsampling x = layers.Conv2DTranspose(64, (2, 2), strides=(2, 2), padding='same')(x) x = layers.Conv2D(64, (3, 3), activation='relu', padding='same')(x) x = layers.BatchNormalization()(x) x = layers.Conv2D(64, (3, 3), activation='relu', padding='same')(x) x = layers.BatchNormalization()(x) x = layers.Conv2D(64, (3, 3), activation='relu', padding='same')(x) x = layers.BatchNormalization()(x) x = layers.Dropout(0.3)(x) x = layers.add([x, residual_2]) x = layers.Conv2DTranspose(32, (2, 2), strides=(2, 2), padding='same')(x) x = layers.Conv2D(32, (3, 3), activation='relu', padding='same')(x) x = layers.BatchNormalization()(x) x = layers.Conv2D(32, (3, 3), activation='relu', padding='same')(x) x = layers.BatchNormalization()(x) x = layers.Conv2D(32, (3, 3), activation='relu', padding='same')(x) x = layers.BatchNormalization()(x) x = layers.Dropout(0.2)(x) x = layers.add([x, residual_1])
- Conv2D Layer: Convolutional layer of the network, where the image pre-processing happens and the kernel filter the image on the whole image, detecting lines and edges.
32
,64
,128
refers to the umber of filters used, starting from a small number.Relu
activation adds non-linearity to the model, to learn more complex patterns.Padding='same
ensures that the output feature map has the same width and height as the input.Batch normalization
is applied after each convolution to stabilize and speed up training. - MaxPooling2D Layer: It performs downsampling operations, and also spatial dimensions (width, height), reducing overfitting and the computional cost by decreasing the spatial dimensionality.
- Dropout Layer: Helps in preventing overfitting by randomly setting a fraction of input units to 0 at each update during training.
- Bottleneck: The central part of the U-Net which doesn't have a skip connection to capture the most abstract features of the image.
- Conv2DTranspose Layer & Residual Connections: The transpose layers, often referred to as deconvolutional layers, help in upsampling the features.
Residual connections
from the downsampling path are added to these upsampled features. This is crucial as these connections help in localizing and refining the segmented regions. - Output Layer: Produces the segmented image. The
sigmoid
activation function ensures pixel values are in the [0,1] range.
- Keras Team. Oxford Pets Image Segmentation. Keras. Retrieved from https://keras.io/examples/vision/oxford_pets_image_segmentation
- GeeksforGeeks Team. How to Normalize, Center, and Standardize Image Pixels in Keras. GeeksforGeeks. Retrieved from https://www.geeksforgeeks.org/how-to-normalize-center-and-standardize-image-pixels-in-keras
- Stack Overflow Community. Image and Mask Normalization in Semantic Segmentation for Cancer. Stack Overflow. Retrieved from https://stackoverflow.com/questions/67699174/image-and-mask-normalization-in-semantic-segmentation-for-cancer
- Analytics Vidhya Team. Getting Started with Image Processing using OpenCV. Analytics Vidhya. Retrieved from https://www.analyticsvidhya.com/blog/2023/03/getting-started-with-image-processing-using-opencv
- OpenCV Documentation (n.d.). Image Filtering. OpenCV Official Documentation. Retrieved from https://docs.opencv.org/4.x/d4/d13/tutorial_py_filtering.html
- SabrePC Blog (n.d.). Understanding the Difference Between Epochs, Batch Size and Iterations. SabrePC Blog. Retrieved from https://www.sabrepc.com/blog/Deep-Learning-and-AI/Epochs-Batch-Size-Iterations
- Brownlee, J. (n.d.). Difference Between a Batch and an Epoch. Machine Learning Mastery. Retrieved from https://machinelearningmastery.com/difference-between-a-batch-and-an-epoch/