The goal of this project is to explore the full end to end life cycle of a Machine learning project. Although the model predictions aren't great, the pipeline to build, train and deploy has been tested to work well.
A live inference demo is deployed on
Train a simple age detection modelConvert trained model to TFLiteUsing Opencv JS & TFJS create a live age detection UIDockerize development environmentDeploy the front end only codeMobile friendly inference- Refactor code base to use async await instead of event listeners?
- Improve age prediction model accuracy
- Explore options to train and use Yolo for face detection
- Age prediction dataset: https://www.kaggle.com/datasets/mariafrenti/age-prediction
- Face detection with Open CV JS: https://docs.opencv.org/3.4/d2/d99/tutorial_js_face_detection.html
- Yolo training with Keras_cv: https://keras.io/examples/vision/yolov8/
- Face detection Dataset: https://www.kaggle.com/datasets/fareselmenshawii/face-detection-dataset/data
- Convert Model to TFLite: https://www.tensorflow.org/lite/models/convert
- Using TFLite with TFJS: https://js.tensorflow.org/api_tflite/0.0.1-alpha.10/
- TFJS Boiler plate copied from Google: https://codelabs.developers.google.com/codelabs/tensorflowjs-object-detection#7
- TFlite Sample Inference: https://codepen.io/jinjingforever/pen/xxgWRaE?editors=1010
- Open CV JS Object detection: https://docs.opencv.org/3.4/df/d6c/tutorial_js_face_detection_camera.html