Skip to content

kriskda/android_ai_cats_vs_dogs

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

7 Commits
 
 
 
 
 
 
 
 
 
 

Repository files navigation

Dogs and cats classification

Sample project with a goal to build and train CNN model and export it to tensorflow lite model to use with the sample android app. It contains:

  • jupyter/ai_model.ipynb - dataset preparation, model optimization and training
  • jupyter/prepare_data.ipynb - not really needed, just to see train/validation images
  • DogsAndCatsAndroid - android sample app. It is missing *.tflite model in data/src/main/ml/ directory. See notes below
ezgif.com-gif-maker.mov

Important notes:

  • Due to size constraints dataset, *.h5 and *.tflite models not provided in this repository
  • There was no motivation to produce nice UI experience. It should just work
  • CNN model was built and trained using tensorflow framework. Validation accuracy reached around 88%

2023-01-30 10 11 43 github com af2b4a2a37de

CNN model architecture. Some parameter values were tuned using keras tuner.

2023-01-30 10 20 16 netron app d6755a2ef91d

About

Simple android app using cats vs dogs AI model

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published