README.md

metadata

license: apache-2.0
language:
  - en
base_model:
  - google/mobilenet_v2_1.0_224
pipeline_tag: image-classification
tags:
  - art
  - imageclassficatio
  - torchvision.models.mobilenet_v2

Image Classification (CPU/GPU)

• Model: torchvision mobilenet_v2 pretrained on ImageNet (Apache-2.0)
• Task: Predict top-5 ImageNet classes for a given image.
• Note: Here we just provide the resources for to run this models in the laptops we didn't develop this entire models we just use the open source models for the experiment this model is developed by TorchVision / PyTorch

Quick start (any project)

# 1) Create env
python -m venv venv && source .venv/bin/activate  # Windows: ./venv/Scripts/activate

# 2) Install deps
pip install -r requirements.txt

# 3) Run
python main.py --help

Tip: If you have a GPU + CUDA, PyTorch will auto-use it. If not, everything runs on CPU (slower but works).

---