Skip to content

Latest commit

 

History

History
57 lines (40 loc) · 1.03 KB

README.md

File metadata and controls

57 lines (40 loc) · 1.03 KB

Image Classifier Django API

Submit an image and get predictions of its content with tensorflow ResNet50 model through an API endpoint.

Example

{
    "id": 3,
    "img": "http://localhost:8000/media/example.jpg",
    "predictions": "Leopard at 97%. Cheetah at 3%. Jaguar at 1%."
}

Installation

Created with python 3.10.6

  • Create Python Virtual Env

    python -m venv env_name

  • Use Python Virtual Env

    source env_name/bin/activate

  • Install requirement

    python -m pip install -r requirements.txt

  • Lauch dev server

    python manage.py runserver

Endpoint

While dev server is launched, you can access http://localhost:8000/api/v1/image-classifier with browser. You can use the debug interface to list and get predictions.

Storage

Images are stored into /media/ folder, predictions data are stored into database. You can configure database with /config/settings.py