Skip to content

Detecting Multiple objects in a video using Single Shot Multibox Detector

Notifications You must be signed in to change notification settings

anushuk/Object-Detection-SSD

Repository files navigation

Object-Detection SSD


Intro

Build Status Python 3.5 Pytorch 0.1.12

Detecting Multiple objects in a video using Single Shot Multibox Detector

Weight Files are splitted as

  • ssd300_mAP_77.43_v2.pth.000
  • ssd300_mAP_77.43_v2.pth.001
  • ssd300_mAP_77.43_v2.pth.002
  • ssd300_mAP_77.43_v2.pth.003
  • ssd300_mAP_77.43_v2.pth.004

Join weight files Here

Read more about SSD here

Click on this image to see demo from SSD:

img

Dependencies

Check out the
virtual_platform_windows.yml file

Getting started

Create a virtual platform.
conda env create -f virtual_platform_windows.yml

Working

[img]

Testing

updateobject_detection.py
reader = imageio.get_reader('video.mp4')

Getting data set for Training a new model

Training is simple but GPU CUDA is mandatory otherwise it will take months to train on a CPU:

#Get the dataset:
(http://host.robots.ox.ac.uk/pascal/VOC/index.html)

Created by:
Anubhav Shukla

lic

About

Detecting Multiple objects in a video using Single Shot Multibox Detector

Topics

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published