sourcery-ai-bot / django-object-detection Goto Github PK

View Code? Open in Web Editor NEW

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

License: Apache License 2.0

JavaScript 1.92% Python 56.93% CSS 0.07% HTML 41.08%

django-object-detection's Introduction

Django Object Detection With YoloV5

Demo of the WebApp

DjangoObjdetection.mov

This is a project to demonstrate easy integration of YoloV5 in Django WebApp and 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 env 

# install dependencies
pip install django
pip install django-crispy-forms
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.