Skip to content

I implement RBM, DBN, Multi-modal DBN with Python and the majority of matrix operations are executed on GPU. In this project I employ Cudamat and Numpy python libs.

vbillys/DeepNet

 
 

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

28 Commits
 
 
 
 
 
 

Repository files navigation

DeepNet

In this project, I implement a deep learning toolbox (DeepNet) including RBM, DBN, Multi-modal DBN with Python, in which the majority of matrix operations are carried on GPU by using the Cudamat to speed up the calculation process. There are some examples to show how to use this package.

This project make some references to the matlab code in https://github.com/dmus/API-Project. However, in comparison with the matlab code, our version improves the performance 25 times (test on the Mnist data).

Requirements

Usage

To use the toolbox, following steps are needed.

(1) compile the Cudamat library :

cd (directory to DeepNet)
cd DeepNet/RBM/cudamat/
Make (note : correct path to gcc-4.6 or below version compiler should be given in Makefile)

(2) change directory to RBM/, then set the DEEPNET_PATH variable in set_env.sh file to the RBM/ path in your computer

(3) run command :

source set_env.sh

(4) We provide some demo programs in this toolbox.

RBM and DBN demos

For RBM and DBN demos, we use Mnist data, which has been contained in our toolbox. To run these demos, you should first uncompress the data in example/.

cd example/
tar -xzvf mnist_data.tar.gz
python rbmDemo.py
or
python DBNdemo.py

For help information, run

python rbmDemo.py --help
or
python DBNdemo.py --help

Multi-modal DBN demo

For multi-modal demo, we employ SHREC 2007 feature data to show the usage. How the data is generated has been elaborated in our paper "Multi-modal Feature Fusion for 3D Shape Recognition and Retrieval". To run this demo, change directory to multi-modal_demo/ and run

python multiModalityDemo.py

For help information, run

python multiModalityDemo.py --help

Platform

This code is only tested on Linux mint-16 64-bit.

About

I implement RBM, DBN, Multi-modal DBN with Python and the majority of matrix operations are executed on GPU. In this project I employ Cudamat and Numpy python libs.

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages

  • C 80.3%
  • Python 13.0%
  • Cuda 4.5%
  • C++ 1.4%
  • Makefile 0.4%
  • QMake 0.2%
  • Other 0.2%