/landmark-detection

Wing Landmark Detection model with image segmentation and detection.

Primary LanguageJupyter Notebook

Wing Landmark Detection

Wing Landmark Detection model with image segmentation and blob detection. Used UNet model architecture to train the model.

Screenshot 2022-12-20 at 12 20 47

Model Architecture and Pipeline

fnins-14-568614-g001

U-Net was used to handle the segmentation tasks.The network aims to process the wing images and output a mask image that has the same size as the input image and consists of pixels that are in the range of 0-1. Annotated wing image coordinates are used to create landmark point masks. After that, those masks were used in training the model.

Screenshot 2022-11-11 at 15 28 25Screenshot 2022-11-11 at 15 28 39

After segmentation, Blob detection method is used to extract coordinates of the landmarks.

Screenshot 2022-11-11 at 15 30 10Screenshot 2022-11-11 at 15 30 25