Giter Club home page Giter Club logo

deepvideoanalytics's Introduction

#Deep Video Analytics โ€ข Build Status 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.

self-promotion: If you are interested in Healthcare & Machine Learning please take a look at my another Open Source project 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 specifications (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 explore.ipynb
  • Some documentation on design decision, architecture and deployment.

Several models included out of the box

We take significant efforts to ensure that following models (code+weights included) work without having to write any code.

Potential algorithms/models

  1. Text detection models
  2. Soundnet (requires extracting mp3 audio)
  3. Pytorch Squeezenet
  4. Mapnet (requires converting models from Marvin)
  5. Keras-js which uses Keras inception for client side indexing

Open Issues & To Do

Installation

Pre-built docker images (corrosponding to alpha version) for both CPU and GPU version are now available on Docker Hub.

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 

Your machine NVidia GPU with Docker and nvidia-docker installed

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.

pip install --upgrade nvidia-docker-compose
git clone https://github.com/AKSHAYUBHAT/DeepVideoAnalytics && cd DeepVideoAnalytics/docker_GPU && ./rebuild.sh && nvidia-docker-compose up 

On AWS EC2 with a GPU enabled P2 instance

We provide an AMI will all dependancies such as docker & drivers. 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.

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 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.

On multiple machines with/without GPUs

Other than the shared media folder (ideally a mounted EFS or NFS), configuring Postgres and RabbitMQ is straightforward. Please read this regarding trade offs.

Options specified via environment variable

Following options can be specified in docker-compose.yml, or your envrionment.

  • ALEX_ENABLE=1 (to use Alexnet with PyTorch. Otherwise disabled by default)
  • YOLO_ENABLE=1 (to use YOLO 9000. Otherwise disabled by default)
  • SCENEDETECT_DISABLE=1 (to disable scene detection, Otherwise enabled by default)

Architecture

Architecture

User Interface

Search

UI Screenshot

Past queries

UI Screenshot

Video list / detail

UI Screenshot UI Screenshot

Frame detail

UI Screenshot

Libraries & Code used

References

Citation for Deep Video Analytics

Coming soon!

Copyright

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

deepvideoanalytics's People

Contributors

akshaybhatai avatar

Watchers

James Cloos avatar Eric avatar

Recommend Projects

  • React photo React

    A declarative, efficient, and flexible JavaScript library for building user interfaces.

  • Vue.js photo Vue.js

    ๐Ÿ–– Vue.js is a progressive, incrementally-adoptable JavaScript framework for building UI on the web.

  • Typescript photo Typescript

    TypeScript is a superset of JavaScript that compiles to clean JavaScript output.

  • TensorFlow photo TensorFlow

    An Open Source Machine Learning Framework for Everyone

  • Django photo Django

    The Web framework for perfectionists with deadlines.

  • D3 photo D3

    Bring data to life with SVG, Canvas and HTML. ๐Ÿ“Š๐Ÿ“ˆ๐ŸŽ‰

Recommend Topics

  • javascript

    JavaScript (JS) is a lightweight interpreted programming language with first-class functions.

  • web

    Some thing interesting about web. New door for the world.

  • server

    A server is a program made to process requests and deliver data to clients.

  • Machine learning

    Machine learning is a way of modeling and interpreting data that allows a piece of software to respond intelligently.

  • Game

    Some thing interesting about game, make everyone happy.

Recommend Org

  • Facebook photo Facebook

    We are working to build community through open source technology. NB: members must have two-factor auth.

  • Microsoft photo Microsoft

    Open source projects and samples from Microsoft.

  • Google photo Google

    Google โค๏ธ Open Source for everyone.

  • D3 photo D3

    Data-Driven Documents codes.