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

GitHub

https://github.com/MotionMobs/Sandwhich