A Machine Learning App for an Age-Old Debate
Sandwhich
Sandwhich is a mobile app built to solve the sandwich debate by determining if the picture you've taken is a sandwich using machine learning, Flutter, and TensorFlow Lite.
Available on Google Play and the App Store
Getting Started
Building from Source
Clone the repository
git clone https://github.com/MotionMobs/Sandwhich
cd Sandwhich
Run Sandwhich
flutter run
Train Your Model/ How To ? Your ?
To train your ?, you need to install Docker and ImageMagick.
brew update && brew install imagemagick
Set up file structure for your images
training_images
│ ├── input
│ │ ├── not-sandwich
│ │ │ ├── processed # processed not-sandwich images
│ │ │ └── unprocessed # original, not yet processed not-sandwich images
│ │ ├── sandwich
│ │ │ ├── processed # processed sandwich imagess
│ │ │ └── unprocessed # original, not yet processed sandwich images
│ ├── output # ready for model
│ │ ├── not-sandwich # minimum 20 images to train
│ │ └── sandwich # minimum 20 images to train
└──
Once original images are loaded into the folders, as described above, process the images
cd training_images
./conversion.sh
After all images have been processed we need to build the Docker image, run the script to retrain the model, and run the script to convert the retrained model to a .tflite model we can use in the app.
Go back to the root of the project and run:
cd -
./train.sh
Prepare to wait a while.
Testing
To test the new model, run ./test_model.sh $IMAGE_PATH where $IMAGE_PATH is any path to an image you want to test with. For convenience, you can even drag an image into the terminal to have its path placed at the end of whatever is currently in the terminal.
- When ready to use the new model, move it to the assets folder and overwrite the old model. Backup old model if desired.
- Run the app, find some sandwiches, and enjoy!
To release on Android:
flutter build apk --target-platform android-arm64