Skip to content

Navigation Menu

Sign in
Appearance settings

Search code, repositories, users, issues, pull requests...

Provide feedback

We read every piece of feedback, and take your input very seriously.

Saved searches

Use saved searches to filter your results more quickly

Appearance settings

rampal-punia/django-object-detection

Open more actions menu

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

Morty Proxy This is a proxified and sanitized view of the page, visit original site.