Skip to content
forked from VIAME/VIAME

Video and Image Analytics for Multiple Environments

License

Notifications You must be signed in to change notification settings

BetsyMcPhail/VIAME

 
 

Repository files navigation

VIAME Logo


VIAME is a computer vision library designed to integrate several image and video processing algorithms together in a common distributed processing framework, majorly targeting marine species analytics. As it contains many common algorithms and compiles several other popular repositories together as a part of its build process, VIAME is also useful as a general computer vision toolkit. The core infrastructure connecting different system components is currently the KWIVER library, which can connect C/C++, python, and matlab nodes together in a graph-like pipeline architecture. Alongside the pipelined image processing system are a number of standalone utilties for model training, output visualization, groundtruth annotation, detector/tracker evaluation (a.k.a. scoring), image/video search, and rapid model generation. Both a desktop application and web version exists to faciliate users who want to work in either environment.

Example Capabilities


Search Example Tracking Example Detection Example

Measurement Example Query Example

Documentation

The User's Quick-Start Guide, Tutorial Videos, and Developer's Manual are more comprehensive, but select entries are also listed below broken down by individual functionality:

Build and Install Guide <> All Examples <> Core Class and Pipeline Info <> Object Detector Examples
GUIs for Visualization and Annotation <> Detector Training API <> Example Plugin Templates
Video Search and Rapid Model Generation <> Scoring and Evaluation of Detectors <> KWIVER Overview
Stereo Measurement Examples <> Embedding Detectors in C++ Code <> How to Integrate Your Own Plugin

Pre-Built Binaries

For a full installation guide see the quick-start slide deck above, but in summary, extract the binaries and place them in a directory of your choosing, for example C:\Program Files\VIAME on Windows or /opt/noaa/viame on Linux. If you're using packages built with GPU support, make sure to have sufficient video drivers installed, version 410.48 or higher. The best way to install drivers depends on your operating system, see below. Lastly, run through some of the examples to validate the installation. It is no longer necessary to install any dependencies of VIAME besides drivers, everything else is packaged inside of it. The binaries are quite large, in terms of disk space, due to the inclusion of multiple default model files and programs, but if just building your desired features from source (e.g. for embedded apps) they are much smaller.

Installation Requirements:
RHEL/CentOS 7 64-Bit, Ubuntu 16.04 64-Bit, Windows 7, 8, or 10 64-Bit
6 Gb of Disk Space for the Full Installation

Installation Recommendations:
NVIDIA Drivers (Version 410.48+ Windows [1] [2] Ubuntu [1] [2] CentOS [1] [2])
A CUDA-enabled GPU with 8 Gb or more VRAM

Linux Desktop Binaries:
VIAME v0.10.2 Ubuntu 16.04/18.04, 64-Bit, GPU Enabled, Python 3.6, Mirror1
VIAME v0.10.2 Ubuntu 16.04/18.04, 64-Bit, GPU Enabled, Python 3.6, Mirror2
VIAME v0.10.2 RHEL/CentOS 7, 64-Bit, GPU Enabled, Python 3.6, Mirror1
VIAME v0.10.2 RHEL/CentOS 7, 64-Bit, GPU Enabled, Python 3.6, Mirror2

Windows Desktop Binaries:
VIAME v0.10.2 Windows 7*/8/10, 64-Bit, GPU Enabled, Python 3.6, Mirror1
VIAME v0.10.2 Windows 7*/8/10, 64-Bit, GPU Enabled, Python 3.6, Mirror2
VIAME v0.10.2 Windows 7*/8/10, 64-Bit, CPU Only, Python 3.6, Mirror1
VIAME v0.10.2 Windows 7*/8/10, 64-Bit, CPU Only, Python 3.6, Mirror2

*Windows 7 requires some updates and service packs installed, e.g. KB2533623.

Web Applications:
VIAME Web Repository (Under Development)
VIAME Deprecated Web Annotator

Optional Patches:
Alternative Generic Detector for IQR Add-On, All OS
Arctic Seals Models Add-On, Linux
Arctic Seals Models Add-On, Windows
HabCam Models (Scallop, Skate, Flatfish) Add-On, Linux
HabCam Models (Scallop, Skate, Flatfish) Add-On, Windows
Low Memory GPU (For 4+ Gb Cards) Add-On, All OS
MOUSS Model Set 1 (Deep 7 Bottomfish) Add-On, All OS
MOUSS Model Set 2 (Deep 7 Bottomfish) Add-On, All OS
MOUSS Sample Project, All Linux

Custom Applications:
Seal Multi-View GUI, CentOS 7, GPU Enabled
Seal Multi-View GUI, Windows 7/8/10, GPU Enabled
Seal Multi-View GUI, Windows 7/8/10, CPU Only

Note: To install Add-Ons and Patches, copy them into an existing VIAME installation folder. To use project files extract them into your working directory of choice. Custom Applications contain a full installation, only with non-default features turned on, and should not be copied into existing installations because they are a full installation and bad things will happen.

Quick Build Instructions

These instructions are intended for developers or those interested in building the latest master ranch. More in-depth build instructions can be found here, but the software can be built either as a super-build, which builds most of its dependencies alongside itself, or standalone. To build VIAME requires, at a minimum, Git, CMake, and a C++ compiler. If using the command line, run the following commands, only replacing [source-directory] and [build-directory] with locations of your choice. While these directories can be the same, it's good practice to have a 'src' checkout then a seperate 'build' directory:

git clone https://github.com/VIAME/VIAME.git [source-directory]

cd [source-directory] && git submodule update --init --recursive

Next, create a build directory and run the following cmake command (or alternatively use the cmake GUI if you are not using the command line interface):

mkdir [build-directory] && cd [build-directory]

cmake -DCMAKE_BUILD_TYPE:STRING=Release [source-directory]

Once your cmake command has completed, you can configure any build flags you want using 'ccmake' or the cmake GUI, and then build with the following command on Linux:

make -j8

Or alternatively by building it in Visual Studio or your compiler of choice on Windows. '-j8' tells the build to run multi-threaded using 8 threads, this is useful for a faster build though if you get an error it can be difficult to see it, in which case running just 'make' might be more helpful. For Windows, currently VS2017 is the desired compiler, though select versions of 2015 and 2019 also work. If using CUDA, version 9.0 and above, with CUDNN 7.0 and above is desired, in particular we recommend CUDA 10. On both Windows and Linux it can also be beneficial to use Anaconda to get multiple standard python packages. Boost Python (turned on by default when Python is enabled) requires Numpy and a few other dependencies.

There are several optional arguments to viame which control which plugins get built, such as those listed below. If a plugin is enabled that depends on another dependency such as OpenCV) then the dependency flag will be forced to on. If uncertain what to turn on, it's best to just leave the default enable and disable flags which will build most (though not all) functionalities. These are core components we recommend leaving turned on:

Flag Description
VIAME_ENABLE_OPENCV Builds OpenCV and basic OpenCV processes (video readers, simple GUIs)
VIAME_ENABLE_VXL Builds VXL and basic VXL processes (video readers, image filters)
VIAME_ENABLE_PYTHON Turns on support for using python processes (multiple algorithms)
VIAME_ENABLE_PYTORCH Installs all pytorch processes (detectors, trackers, classifiers)

And a number of flags which control which system utilities and optimizations are built, e.g.:

Flag Description
VIAME_ENABLE_CUDA Enables CUDA (GPU) optimizations across all processes (PyTorch, etc...)
VIAME_ENABLE_CUDNN Enables CUDNN (GPU) optimizations across all processes
VIAME_ENABLE_VIVIA Builds VIVIA GUIs (tools for making annotations and viewing detections)
VIAME_ENABLE_KWANT Builds KWANT detection and track evaluation (scoring) tools
VIAME_ENABLE_DOCS Builds Doxygen class-level documentation for projects (puts in install share tree)
VIAME_BUILD_DEPENDENCIES Build VIAME as a super-build, building all dependencies (default behavior)
VIAME_INSTALL_EXAMPLES Installs examples for the above modules into install/examples tree
VIAME_DOWNLOAD_MODELS Downloads pre-trained models for use with the examples and training new models

And lastly, a number of flags which build algorithms with more specialized functionality:

Flag Description
VIAME_ENABLE_SMQTK Builds SMQTK plugins for image/video search
VIAME_ENABLE_SCALLOP_TK Builds Scallop-TK based object detector plugin
VIAME_ENABLE_YOLO Builds YOLO (Darknet) object detector plugin
VIAME_ENABLE_BURNOUT Builds Burn-Out based pixel classifier plugin
VIAME_ENABLE_ITK Builds ITK cross-modality image registration
VIAME_ENABLE_UW_CLASSIFIER Builds UW fish classifier plugin
VIAME_ENABLE_TENSORFLOW Builds TensorFlow object detector plugin
VIAME_ENABLE_SEAL_TK Builds Seal Multi-Modality GUI
VIAME_ENABLE_MATLAB Turns on support for and installs all matlab processes
VIAME_ENABLE_LANL Builds an additional (Matlab) scallop detector

Source Code Layout

 VIAME
   ├── cmake               # CMake configuration files for subpackages
   ├── doc                 # Documentation files and manual (pre-compilation)
   ├── configs             # All system-runnable config files and models
   │   ├── pipelines       # All processing pipeline configs
   │   │   └── models      # All models, which only get downloaded based on flags
   │   ├── prj-linux       # Default linux project files
   │   └── prj-windows     # Default windows project files 
   ├── examples            # All runnable examples and example tutorials
   ├── packages            # External projects used by the system
   │   ├── kwiver          # Processing backend infastructure
   │   ├── fletch          # Dependency builder for things which don't change often
   │   ├── kwant           # Scoring and detector evaluation tools
   │   ├── vivia           # Baseline desktop GUIs (v1.0)
   │   └── ...             # Assorted other packages (typically for algorithms)
   ├── plugins             # Integrated algorithms or wrappers around external projects
   │   └── ...             # Assorted plugins (detectors, depth maps, filters, etc.)
   └── README.md           # Project introduction page that you are reading

Update Instructions

If you already have a checkout of VIAME and want to switch branches or update your code, it is important to re-run:

git submodule update --init --recursive

After switching branches to ensure that you have on the correct hashes of sub-packages within the build. Very rarely you may also need to run:

git submodule sync

Just in case the address of submodules has changed. You only need to run this command if you get a "cannot fetch hash #hashid" error.

License, Citations, and Acknowledgements

VIAME is released under a BSD-3 license.

A non-exhaustive list of relevant papers used within the project alongside contributors can be found here.

VIAME was developed with funding from multiple sources, with special thanks to those listed here.

About

Video and Image Analytics for Multiple Environments

Resources

License

Stars

Watchers

Forks

Packages

No packages published

Languages

  • C++ 40.5%
  • Python 36.6%
  • CMake 8.9%
  • MATLAB 7.1%
  • HTML 2.8%
  • Shell 2.0%
  • Other 2.1%