Skip to content

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

About

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

Topics

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Contributors 2

  •  
  •  

Languages