Skip to content

Django object detection app using yolov5. Upload new custom model or use any of the yolov5 pre-trained model.

License

Notifications You must be signed in to change notification settings

rampal-punia/django-object-detection

Repository files navigation

Django Object Detection With YoloV5

Demo of the WebApp

DjangoObjdetection.mov

A project to demonstrate easy integration of YoloV5 in Django WebApp

Note: This is not a full-fledged production ready app though can be scaled to work as one.

Features of the WebApp

  • Create/Edit ImageSets.
  • Upload multiple images with dropzonejs to the selected ImageSet.
  • Convert uploaded image size to 640 x 640. (For faster detection)
  • Upload/update a custom pre-trained model.(If you have offline files of a model)
  • YoloV5 models will download upon selection. (Active internet connection required for this step.)
  • Detect object on an image with YoloV5/custom pre-trained model.

Note

An image with the name default.png in media folder is required for user-profile. Create media folder and add any image file with this name 'default.png'.

Steps to use locally

clone the repo locally

create virtual environment

# install dependencies
pip install django
pip install django-crispy-forms
pip install crispy-bootstrap4
pip install django-cleanup
pip install django-debug-toolbar
pip install celery
pip install yolov5

# migrate
python manage.py migrate

# create super user
python manage.py createsuperuser # (it may show an error page if no 'default.png' in media folder. See note above.)

# run
python manage.py runserver

login
# Login at the web address 127.0.0.1:8000 using the superuser credentials.

Create ImageSet
# create an ImageSet first and then upload images into the ImageSet from ImageSet detail page.

# On images list page click on detect object.

# select a YoloV5 model
# the YoloV5 dependencies and pre-trained model will start downloading.

Apps

  • Detectobj
  • images
  • modelmanager
  • users

Javascript library

  • dropzonejs
  • ekko-lightbox

Django starter template used

DjangoAdvancedBoilerplate

Releases

No releases published

Packages

No packages published

Languages