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:
Check out the
virtual_platform_windows.yml file
Create a virtual platform.
conda env create -f virtual_platform_windows.yml
updateobject_detection.py
reader = imageio.get_reader('video
.mp4')
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