Ecosyste.ms: Awesome

An open API service indexing awesome lists of open source software.

Awesome Lists | Featured Topics | Projects

https://github.com/npatta01/pytorch-food

Last synced: 14 days ago
JSON representation

Host: GitHub
URL: https://github.com/npatta01/pytorch-food
Owner: npatta01
Created: 2021-09-13T01:01:40.000Z (over 3 years ago)
Default Branch: initial
Last Pushed: 2021-11-19T22:55:23.000Z (about 3 years ago)
Last Synced: 2024-12-07T07:04:31.008Z (18 days ago)
Language: Jupyter Notebook
Size: 23.8 MB
Stars: 3
Watchers: 3
Forks: 0
Open Issues: 0
Metadata Files:
- Readme: readme.md

Awesome Lists containing this project

README

# README

## About

This repo is a tutorial/sample code for :
1. How to train a computer vision classifier using PyTorch
2. Serve the model using PyTorch serving
3. Deploy the model using Torch Mobile for Android
4. Deploy the model using Torch Mobile for IOS

## Dataset

This project uses the [Food-101](https://data.vision.ee.ethz.ch/cvl/datasets_extra/food-101/] dataset.

## Demo

### Android
![android](assets/android.png)

### iOS

![ios](assets/ios.png)

## References

The android and ios code was forked from the very helpful examples provided by the Pytorch Team.
[IOS](https://github.com/pytorch/ios-demo-app)
[Android](https://github.com/pytorch/android-demo-app)

## Notes

```
cd notebooks
papermill -p model resnet34 01_training.ipynb 01_training__output.ipynb -k python3

```