Skip to content

chagge/DeepVideoAnalytics

 
 

Repository files navigation

#Deep Video Analytics • Build Status

Akshay Bhat, Cornell University. Website & Contact

Banner Deep Video Analytics provides a platform for indexing and extracting information from videos and images. Deep learning detection and recognition algorithms are used for indexing individual frames / images along with detected objects. The goal of Deep Video analytics is to become a quickly customizable platform for developing visual & video analytics applications, while benefiting from seamless integration with state or the art models released by the vision research community.

Advertisement: If you are interested in Healthcare & Machine Learning please take a look at Computational Healthcare

Features

  • Visual Search using Nearest Neighbors algorithm as a primary interface
  • Upload videos, multiple images (zip file with folder names as labels)
  • Provide Youtube url to be automatically processed/downloaded by youtube-dl
  • Metadata stored in Postgres
  • Operations (Querying, Frame extraction & Indexing) performed using celery tasks and RabbitMQ
  • Separate queues and workers for selection of machines with different spect (GPU vs RAM)
  • Videos, frames, indexes, numpy vectors stored in media directory, served through nginx
  • Explore data, manually run code & tasks without UI via a jupyter notebook (e.g. explore.ipynb)
  • A set of notes are available for guidance regarding multi-machine deployment and design decisions.

Implemented & Potential algorithms/models

Installation

On Mac, Windows and Linux machines without NVidia GPUs

You need to have latest version of Docker installed.

git clone https://github.com/AKSHAYUBHAT/DeepVideoAnalytics && cd DeepVideoAnalytics/docker && docker-compose up 
```
### Installation for machines with GPU or AWS EC2 P2.xlarge instances

#### Machine with GPU
Replace docker-compose by nvidia-docker-compose, the Dockerfile uses tensorflow gpu base image and appropriate version of pytorch. The Makefile for Darknet is also modified accordingly. This code was tested using an older NVidia Titan GPU and nvidia-docker.

```bash
pip install --upgrade nvidia-docker-compose
git clone https://github.com/AKSHAYUBHAT/DeepVideoAnalytics && cd DeepVideoAnalytics/docker_GPU && nvidia-docker-compose up 
```
#### AWS EC2 AMI
Start a P2.xlarge instance with **ami-b3cc1fa5** (N. Virginia), ports 8000, 6006, 8888 open (preferably to only your IP) and run following command after ssh'ing into the machine. 
```bash
cd deepvideoanalytics && git pull && cd docker_GPU && ./rebuild.sh && nvidia-docker-compose up 
```
you can optionally specify "-d" at the end to detach it, but for the very first time its useful to read how each container is started. After approximately 3 ~ 5 minutes the user interface will appear on port 8000 of the instance ip.
The Process used for [AMI creation is here](https://github.com/AKSHAYUBHAT/DeepVideoAnalytics/blob/master/notes/ami.md) **Security warning!** The current GPU container uses nginx <-> uwsgi <-> django setup to ensure smooth playback of videos. 
However it runs nginix as root (though within the container). Considering that you can now modify AWS Security rules on-the-fly, I highly recommend allowing inbound traffic only from your own IP address.

### Deployment on multiple machines with/without GPUs

Please read this document for [guidance on multi-machine deployment](https://github.com/AKSHAYUBHAT/DeepVideoAnalytics/blob/master/notes/architecture.md).
 
## Architecture overview
![Architecture](demo/architecture.png "System architecture")

## User Interface 
### Search
![UI Screenshot](demo/search.png "search")
### Past queries
![UI Screenshot](demo/past_query.png "past queries")
### Video list / detail
![UI Screenshot](demo/video_list.png "Video list")
![UI Screenshot](demo/video_detail.png "detail")
### Frame detail
![UI Screenshot](demo/frame_detail.png "Frame detail")

## Libraries & Code used
** If I have missed anything please email me. **

- Pytorch [License](https://github.com/pytorch/pytorch/blob/master/LICENSE)
- Darknet [License](https://github.com/pjreddie/darknet/blob/master/LICENSE)
- AdminLTE2 [License](https://github.com/almasaeed2010/AdminLTE/blob/master/LICENSE)
- FabricJS [License](https://github.com/kangax/fabric.js/blob/master/LICENSE)
- Modified PySceneDetect [License](https://github.com/Breakthrough/PySceneDetect)
- Docker 
- Nvidia-docker
- OpenCV
- Numpy
- FFMPEG
- Tensorflow

# Copyright
**Copyright 2016-2017, Akshay Bhat, Cornell University, All rights reserved.**

About

Analyze videos & images, perform detections, index frames & detected objects, search by examples.

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages

  • JavaScript 59.9%
  • HTML 22.9%
  • C 9.8%
  • CSS 4.8%
  • Python 1.6%
  • Cuda 0.9%
  • Other 0.1%