Skip to content

mhazoglou/PVM_PyCUDA

Repository files navigation

PVM_PyCUDA

An CUDA based optimization of the Predictive Vision Model by Piekniewski et. al. in https://arxiv.org/abs/1607.06854 as well as some extensions that I personally used in my research.

Getting Started

After installing the dependencies go through the jupyter notebook Quick_Start_Tutorial_PVM.ipynb to download data and train your first PVM.

Prerequisites

CUDA 8 (Future versions may support CUDA 9 and 10)

PyCUDA

OpenCV

NumPy

MatPlotLib

Installing

This module relies on PyCUDA which needs to be installed before anything else can function. All other dependencies are pip/conda installable. So far the CUDA code uses CUDA 8, for it to run on newer versions of CUDA you need to change the reductions in UsefulKernels.cu

About

A CUDA based implementation of the Predictive Vision Model

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published