https://github.com/stefanbschneider/pytorch-django
🔥📷 A simple Django web app classifying images using PyTorch.
https://github.com/stefanbschneider/pytorch-django
classification densenet django heroku image-classification imagenet python pytorch vision web-application
Last synced: about 2 months ago
JSON representation
🔥📷 A simple Django web app classifying images using PyTorch.
- Host: GitHub
- URL: https://github.com/stefanbschneider/pytorch-django
- Owner: stefanbschneider
- License: mit
- Created: 2020-12-30T16:48:44.000Z (over 4 years ago)
- Default Branch: main
- Last Pushed: 2022-02-20T14:17:20.000Z (about 3 years ago)
- Last Synced: 2025-03-18T17:57:02.082Z (2 months ago)
- Topics: classification, densenet, django, heroku, image-classification, imagenet, python, pytorch, vision, web-application
- Language: Python
- Homepage: https://pytorch-django.herokuapp.com/
- Size: 16.3 MB
- Stars: 38
- Watchers: 2
- Forks: 10
- Open Issues: 0
-
Metadata Files:
- Readme: README.md
- License: LICENSE
Awesome Lists containing this project
README
# Django + PyTorch Image Classification App
A simple Django web app classifying uploaded images using a pretrained PyTorch DenseNet.
As an example of how to integrate PyTorch with Django and deploy it on Heroku.
* Live demo: [Heroku](https://pytorch-django.herokuapp.com/) (might take multiple seconds to start the free Heroku instance)
* Corresponding blog post: [Using PyTorch Inside a Django App](https://stefanbschneider.github.io/blog/pytorch-django)
## Setup
```
pip install -r requirements
```
## Usage
Live demo: [Heroku](https://pytorch-django.herokuapp.com/) (might take multiple seconds to start the free Heroku instance)
### Development
```
python manage.py runserver
```
The app is running on http://localhost:8000/
### Production Deployment on Heroku
See description in [blog post](https://stefanbschneider.github.io/blog/pytorch-django).