Image_classification

About   |   Features   |   Technologies   |   Requirements   |   Starting   |   License   |   Author

🎯 About

This is a simple image classification project to classify siim dataset. In this project we train two models Alexnet and Resnet18 models with both pretrained amd non-pretrained weights. This project is based from Approaching Any Machine Learning problem boom by Abhishek Thakur.

🚀 Technologies

The following tools were used in this project:

• Python
• Pytorch

✅ Requirements

Before starting 🏁, you need to have Git and Python, Pytorch installed.

🏁 Starting

# Clone this project
$ git clone https://github.com/chandramouli224/image_classification

# Access
$ cd image_classification

# Create Virtual Environment
$ python -m venv venv

# Activate virtual Environment
$ .\venv\Scripts\activate

# Install Dependencies
$ pip install -r requirements.txt

# Run the project
$ python src/train.py

Made with ❤️ by Chandramouli Guna

 

Back to top