This project is trained on top of a pre-trained MobileNet model(used to classify 98 different objects) to generate a new model that detects sign languages(5 signs). Transfer learning is advantageous as it addresses the known challanges while training a CNN model which require(longer training time, a large dataset and exceptionally advanced computing power). The project utilizes only 80 images for the additional training and only takes 30 minutes to train. The project does the following:
- Collects training and testing images.
- Label the images and split them in to training and testing sets.
- Creates a label map.
- Generates Tensorflow Records.
- Modifies the MobileNet model(Transfer learning).
- Train the Model.
- Detect in real time.
- Clone this project
- Follow the instructions here to install the Tensorflow Object Detection API and the pre-trained model - https://tensorflow-object-detection-api-tutorial.readthedocs.io/en/latest/install.html
- Clone the labelImg repository - https://github.com/heartexlabs/labelImg
- Follow the instructions in the jupyter notebooks Data collector.ipynb to capture images and Object Detector.ipynb to train your model.