Skip to content

tiffanny29631/dash-object-detection

 
 

Repository files navigation

Dash Object Detection Explorer

This is a demo of the Dash interactive Python framework developed by Plotly.

Dash abstracts away all of the technologies and protocols required to build an interactive web-based application and is a simple and effective way to bind a user interface around your Python code. To learn more check out our documentation.

Try out the demo app here.

Animated1

Getting Started

Using the demo

To get started, select a footage you want to view, and choose the display mode (with or without bounding boxes). Then, you can start playing the video, and the visualization will be displayed depending on the current time.

Running the app locally

First create a virtual environment with conda or venv inside a temp folder, then activate it.

virtualenv dash-object-detection

# Windows
dash-object-detection\Scripts\activate
# Or Linux
source venv/bin/activate

Clone the git repo, then install the requirements with pip

git clone https://github.com/plotly/dash-object-detection.git
cd dash-object-detection
pip install -r requirements.txt

Run the app

python app.py

About the app

The videos are displayed using a community-maintained Dash video component. It is made by two Plotly community contributors. You can find the source code here.

All videos used are open-sourced under Creative Commons. The original links can be found here.

Model

The object detection model is the MobileNet v1, made by Google and trained on the COCO dataset. You can find their implementation on their official Github repo. You are encouraged to try this app with other models.

Bounding Box Generation

The data displayed in the app are pregenerated for demo purposes. To generate the csv files containing the objects detected for each frame, as well as the output video with bounding boxes, please refer to utils/generate_video_data.py. You will need the latest version of tensorflow and OpenCV, as well as the frozen graph ssd_mobilenet_v1_coco, that you can download in the Model Zoo. Make sure to place the frozen graph inside the same folder as generate_video_data.py, i.e. utils.

Built With

  • Dash - Main server and interactive components
  • Plotly Python - Used to create the interactive plots
  • OpenCV - Create the video with bounding boxes
  • Tensorflow - Generate the bounding box data

Contributing

Please read CONTRIBUTING.md for details on our code of conduct, and the process for submitting pull requests to us.

Authors

See also the list of contributors who participated in this project.

License

This project is licensed under the MIT License - see the LICENSE.md file for details

Screenshots

Screenshot1 Screenshot2

About

Dash Demo App - Object Detection Application, using MobileNet v1

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages

  • Python 77.5%
  • CSS 22.5%