Image Classification App

This project is a simple web application for image classification built using Django and PyTorch. It uses a pretrained DenseNet model to classify uploaded images based on ImageNet labels.

Features

✳️Upload images via a web interface.

✳️Classify images using a pretrained PyTorch DenseNet model.

✳️Responsive UI with Bootstrap 5.

✳️Secure handling of image uploads (images are not saved on the server).

Requirements

Ensure you have the following installed:

💠Python 3.8 or higher

💠Django 4.0 or higher

💠PyTorch 1.9 or higher

💠TorchVision

💠Pillow (for image handling)

Project Structure

project/ ├── image_classification/

│ ├── migrations/

│ ├── static/

│ ├── templates/

│ ├── init.py

│ ├── admin.py

│ ├── apps.py

│ ├── models.py

│ ├── tests.py

│ ├── urls.py

│ ├── views.py

├── project_name/

│ ├── init.py

│ ├── asgi.py

│ ├── settings.py

│ ├── urls.py

│ ├── wsgi.py

├── manage.py

├── requirements.txt

└── README.md

Dependencies

This project uses the following libraries:

✾Django

✾PyTorch

✾TorchVision

✾Pillow

Troubleshooting

If you encounter errors such as missing modules, ensure all dependencies are installed correctly by running:

pip install -r requirements.txt

If you face any issues with PyTorch installation, refer to the official PyTorch installation guide.

Contribution

Contributions are welcome! Feel free to fork this repository and submit pull requests.