Skip to content

Object detection using Single-Shot-Detection architecture using MobileNet as the basenet

License

Notifications You must be signed in to change notification settings

abhileshborode/SSD-MobileNet

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

3 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

Object Detection - SSD-MobileNet

  • Learn about MobileNets and separable depthwise convolutions.
  • The SSD (Single Shot Detection) architecture used for object detection
  • Use pretrained TensorFlow object detection inference models to detect objects
  • Use different architectures and weigh the tradeoffs.
  • Apply an object detection pipeline to a video.

Open the notebook and work through it!

Requirements

Install environment with Anaconda:

conda env create -f environment.yml

Change TensorFlow pip installation from tensorflow-gpu to tensorflow if you don't have a GPU available.

The environment should be listed via conda info --envs:

# conda environments:
#
carnd-advdl-odlab        /usr/local/anaconda3/envs/carnd-advdl-odlab
root                  *  /usr/local/anaconda3

Further documentation on working with Anaconda environments.

Particularly useful sections:

https://conda.io/docs/using/envs.html#change-environments-activate-deactivate https://conda.io/docs/using/envs.html#remove-an-environment

Resources

Future Work

  • Experiment with Fast-R-CNN / Faster-R-CNN and compare its performance with SSD on the same * Driving video

About

Object detection using Single-Shot-Detection architecture using MobileNet as the basenet

Topics

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published