Skip to content
Python framework that facilitates the quick development of complex video analysis applications and other series-processing based applications in a multiprocessing environment.
Branch: master
Clone or download
jadielam Merge pull request #79 from jadielam/master
Added mask image transformer
Latest commit 9eaf708 Jun 21, 2019
Permalink
Type Name Latest commit message Commit time
Failed to load latest commit information.
assets Added ability to control speed of identity processor Jun 4, 2019
docs Revert "Creating the basis for pose estimation on cpu and other impro… Jun 20, 2019
examples
temp Added mask image transformer Jun 21, 2019
tests Fixed bug of blocking situation when task was the last one in the top… Jun 18, 2019
videoflow Added mask image transformer Jun 21, 2019
.gitignore Revert "Creating the basis for pose estimation on cpu and other impro… Jun 20, 2019
.travis.yml Added tests to check that resources are present May 27, 2019
CONTRIBUTING.md comitting -> committing May 25, 2019
Dockerfile changed path of build May 26, 2019
HN.md Create HN.md May 27, 2019
LICENSE First commit Apr 15, 2019
README.md Revert "Creating the basis for pose estimation on cpu and other impro… Jun 20, 2019
ROADMAP.md Update ROADMAP.md Jun 6, 2019
publish.sh Added documentation on how to install videoflow May 13, 2019
setup.py Small changes to documentation Jun 4, 2019

README.md

Videoflow

Videoflow

Build Status license

Videoflow is a Python framework for video stream processing. The library is designed to facilitate easy and quick definition of computer vision stream processing pipelines. It empowers developers to build applications and systems with self-contained Deep Learning and Computer Vision capabilities using simple and few lines of code. It contains off-the-shelf reference components for object detection, object tracking, human pose estimation, etc, and it is easy to extend with your own.

The complete documentation to the project is located in docs.videoflow.dev

Installing the framework

Requirements

Before installing, be sure that you have cv2 and tensorflow >= 1.12 already installed. Python 2 is NOT SUPPORTED. Requires Python 3.6+. There are some known issues to run it on Windows too

Installation

You can install directly using pip by doing pip3 install videoflow

Alternatively, you can install by:

  1. Clone this repository
  2. Inside the repository folder, execute pip3 install . --user

Usage with docker

# clone repo
docker build -t repo/videoflow:latest .
# runs examples/object_detector.py by default
docker run -u $(id -u):$(id -g) -v $(pwd):/usr/src/app repo/videoflow
# or mount the volume from your code directory  to /usr/src/app
docker run -u $(id -u):$(id -g) -v $(pwd):/usr/src/app repo/videoflow python /usr/src/app/yourown.py

Contributing:

A tentative roadmap of where we are headed.

Contribution rules.

If you have new processors, producers or consumers that you can to create, check the videoflow-contrib project. We want to keep videoflow succinct, clean, and simple, with as minimal dependencies to third-party libraries as necessaries. videoflow-contrib is better suited for adding new components that require new library dependencies.

Sample Videoflow application:

Below a sample videoflow application that detects automobiles in an intersection. For more examples see the examples folder. It uses detection model published by tensorflow/models

IMAGE ALT TEXT HERE

import videoflow
import videoflow.core.flow as flow
from videoflow.core.constants import BATCH
from videoflow.consumers import VideofileWriter
from videoflow.producers import VideofileReader
from videoflow.processors.vision.detectors import TensorflowObjectDetector
from videoflow.processors.vision.annotators import BoundingBoxAnnotator
from videoflow.utils.downloader import get_file

URL_VIDEO = "https://github.com/videoflow/videoflow/releases/download/examples/intersection.mp4"

class FrameIndexSplitter(videoflow.core.node.ProcessorNode):
    def __init__(self):
        super(FrameIndexSplitter, self).__init__()
    
    def process(self, data):
        index, frame = data
        return frame

input_file = get_file("intersection.mp4", URL_VIDEO)
output_file = "output.avi"
reader = VideofileReader(input_file)
frame = FrameIndexSplitter()(reader)
detector = TensorflowObjectDetector()(frame)
annotator = BoundingBoxAnnotator()(frame, detector)
writer = VideofileWriter(output_file, fps = 30)(annotator)
fl = flow.Flow([reader], [writer], flow_type = BATCH)
fl.run()
fl.join()

The output of the application is an annotated video:

The Structure of a flow application

A flow application usually consists of three parts:

  1. In the first part of the application you define a directed acyclic graph of computation nodes. There are 3 different kinds of nodes: producers, processors and consumers. Producer nodes create data (commonly they will get the data from a source that is external to the flow). Processors receive data as input and produce data as output. Consumers read data and do not produce any output. You usually use a consumer when you want to write results to a log file, or when you want to push results to an external source (rest API, S3 bucket, etc.)

  2. To create a flow object, you need to pass to it your list of producers and your list of consumers. Once a flow is defined you can start it. Starting the flow means that the producers start putting data into the flow and processors and consumers start receiving data. Starting the flow also means allocating resources for producers, processors and consumers. For simplicity for now we can say that each producer, processor and consumer will run on its own process space.

  3. Once the flow starts, you can also stop it. When you stop the flow, it will happen organically. Producers will stop producing data. The rest of the nodes in the flow will continue running until the pipes run dry. The resources used in the flow are deallocated progressively, as each node stops producing/processing/consuming data.

You can’t perform that action at this time.