ImageClassification

About

This image classification tool uses HuggingFace's timm library with a ResNeXt model to perform image classification.

Click here to view the list of ImageNet class mappings to see what this model can classify.

Installation

1. Ensure you have Python installed

2. Clone this repository to your local machine using the following command:

git clone https://github.com/samnoyes/ImageClassification.git

3. Install the required dependencies. In terminal: navigate to the project directory and run the following command:

pip install -r requirements.txt (pip3 if you are using python3)

This will install the necessary packages.

Usage

Find an image you would like to classify, and pass the URL as a string argument when you run the script:

python image_classification.py "<image url>" (or python3 depending on your version)

Sample Output:

Input: python image_classification.py "https://a-z-animals.com/media/animals/images/original/lion7.jpg"

lion 0.9996147155761719
chow 0.000105000362964347
tiger 5.202060492592864e-05
leopard 3.0612063710577786e-05
cheetah 2.1639469196088612e-05

Credit

Got a lot of help from the HuggingFace docs: https://huggingface.co/docs/timm/models/resnext