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.
Building from Source
Clone the repository
git clone https://github.com/MotionMobs/Sandwhich cd Sandwhich
Train Your Model/ How To 🚋 Your 🥪
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.
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