Skip to content

Latest commit

 

History

History
152 lines (113 loc) · 8.38 KB

README.md

File metadata and controls

152 lines (113 loc) · 8.38 KB
Raw

Machine_learning_projects

This repository contains my projects relating to Machine Learning. To view my notebook directly from Github, you can use nbviewer. Belows are brief description and demonstration of my projects:

  1. Facemask detection with Deeplearning and OpenCV
  2. Traffic signs detection with YOLOv4
  3. Gun detection with YOLOv3
  4. Image caption generator
  5. License Plate Detection and Recognition
  6. Car brand classification with EfficientNet (70% accuracy)
  7. Object classisfication with YOLOv3
  8. Detect COVID-19 from X-ray images
  9. Traffic signs classification with German Traffic Sign dataset on Kaggle (97% accuracy)
  10. Age prediction
  11. Dog and cat classification (95% accuracy)
  12. Fashion classification with fashion_mnnist datasets from Keras (91% accuracy)

Facemask detection with Deeplearning and OpenCV

In this project, I have conducted several methods as below:

  • Train a model to classifiy "No mask" and "Mask" status with MobileNets. Accuracy > 98%
  • Use pre-train Face detection model to catch faces


Traffic signs detection with YOLOv4

  • The YOLOv4 traffic sign detector was trained with darknet and deployed with OpenCV. This article demonstrates more details about the procedure used to train this detector.


Gun detection with YOLOv3:

  • Dataset: 3k images of hand gun and their label annotation (bounding box coordination)
  • Model was trained on Colab with 900 epochs using Yolov3.



Image caption generator:

  • The idea of this project is to create a Deep Learning model that deliver textual description from given photographs. Thus a combination of different techniques from both Computer Vision and Natural Langue Processing are conducted.
  • I used pre-trained weight on imagenet dataset of Resnet50 architecture to extract training features. Later LSTM and Reset50 are combined as one deep CNN to train this caption generator model.
  • The dataset used for this dataset can be downloaded from Kaggle.

License Plate Detection and Recognition:

  • In this project, I created a pipeline that detect and read car license plate in Europe. The piple contains three main statges:

  1. Detect and crop the region of License Plate from rear care image using WPOD model

  2. Using common OpenCV techniques such as blur, threshold, findContour to crop all Plate's digits

  3. Using a Nvidia model which was trained to recognize plate digits.


Car brand classification with EfficientNet

  • Dataset: Stanford car dataset which contains of 16k images from 196 classes of car brands
  • Model was trained with fine-tune pre-trained EfficientNet and acquire the accuracy of 73%.


Object classification with YOLOv3:

  • Dataset: COCO dataset is a large-scale object-detection dataset which contains of approximately 330K images.
  • In this project, I applied YOLOv3 algorithm on the dataset of COCO and pass an image and video through. The results can be seen as below figures:


Detect COVID-19 from X-ray images

The dataset of this project is share by Dr. Joseph Pau Cohen from University of Montreal in an attempt to predict and understand the infection. His repository where I downloaded the dataset can be found here.

Below are some key points regarding this project:

  1. The model was train in the dataset of 25 X-ray images of. Covid patients and 25 X-ray images of normal patients.
  2. The model uses Resnet50 as transfer learning model.
  3. The accuracy of this model is 90%.
  4. The result are for reference only.


Traffic signs classification with German Traffic Sign dataset on Kaggle:

  • Dataset: The Kaggle German Traffic Sign dataset consists of 40 classes and more than 50,000 images in total.
  • The mode created in this project contains 478,763 parameter and was trained for 20 epochs. The final accuracy after testing with new test data is 96.8%


Age prediction:

  • In this project, I trained my a Age classifier with EfficientNet which was claimed to achiever better accuracy than than previous ConvNets. I applied transfer learning with imagenet pre-trained weight and add custom layer. During training, I noticed a significant over-fitting trend, thus I added more Dropout layer and apply several data augmentation techniques.

  • I have created my own dataset consist of 51048 images divided among 9 age groups. However during training process, I realize my dataset is not a good training source since it didn't distributed evenly among age groups. In particularly, my classifier is bias toward the age group of 33-45 and 60 above.

  • I also realized a noticeable pattern which can be related to humans' common sens in age classifying. My classifier considers features such as lighter skin, no beard, black hair, etc as indications for younger age while features such as wrinkles, white hair and beards, etc usually return older age-group results.

  • After long hours of training and tuning, I can only manage to achieve maximum 50% accuracy on my validation set, which is not a good result. I also tried with other popular networks as MobileNets, VGG16, ResNet50, Xcepticon but the accuracy were much worse than EfficientNet.

  • My dataset can be downloaded from here.



Dog and Cat classification:

  • Dataset: this datset contains 25,000 images of dog and cat collected from Internet.
  • My customized model contains 12 layers had achieved accuracy 87.8%. Used transfer learning with pre-trained Resnet50, I have improved my model accuracy up to nearly 98%.


Fashion classification with fashion_mnnist datasets from Keras:

  • Dataset: The fashion_mnist datasets of Keras contains of 60,000 examples and a test set of 10,000 examples
  • My model consists of 597,786 parameters and was trained for 10 epochs. The accuracy is nearly 95%