Skip to content

End-to-end machine learning project showcasing age prediction with a comprehensive pipeline, including TFLite conversion, live UI with TensorFlow.js, and Dockerized development environment.

Notifications You must be signed in to change notification settings

Infinage/Age-Prediction

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

10 Commits
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

Age Detection

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

  1. Netlify.
  2. AWS with custom Subdomain.

TODO

  1. Train a simple age detection model
  2. Convert trained model to TFLite
  3. Using Opencv JS & TFJS create a live age detection UI
  4. Dockerize development environment
  5. Deploy the front end only code
  6. Mobile friendly inference
  7. Refactor code base to use async await instead of event listeners?
  8. Improve age prediction model accuracy
  9. Explore options to train and use Yolo for face detection

Useful links

About

End-to-end machine learning project showcasing age prediction with a comprehensive pipeline, including TFLite conversion, live UI with TensorFlow.js, and Dockerized development environment.

Topics

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages