/object-detection-api

Integrate ML/DL models(Object Detection yolov3) with Flask API for easy & interactive web Interface.

Primary LanguagePython

Object Detection Flask API

Custom object detection PyTorch yoloV3 pretrained model(Drone Detection)

Installation

Python Version

  • Python == 3.7

Library Installation

Windows

  • Virtual Environment
    • python -m venv venv
    • .\venv\Scripts\activate
    • If any problem for scripts activation
      • Execute following command in administration mode
        • Set-ExecutionPolicy Unrestricted -Force
      • Later you can revert the change
        • Set-ExecutionPolicy restricted -Force
  • Library Install
    • python .\installation\get-pip.py
    • pip install --upgrade pip
    • pip install --upgrade setuptools
    • pip install -r requirements.txt

Linux

  • Virtual Environment
    • python -m venv venv
    • source venv/bin/activate
  • Library Install
    • pip install --upgrade pip
    • pip install --upgrade setuptools
    • pip install -r requirements.txt

Pretrained Weight Download

  • Download pretrained model from Follwing Link:
  • Place in the ml/weights folder

Drone Detection

Web Interface View

Male.jpg
Male.jpg

Detection from Web Interface

  • Run flask app
    • python main.py
  • Open web interface
  • Upload image from anywhere
    • Allowed image extensions [jpg, png, jpeg]

Referecne