Coder Social home page Coder Social logo

walkerlau / gpu-cnn Goto Github PK

View Code? Open in Web Editor NEW
11.0 4.0 9.0 11.52 MB

Accelerate convolution neural network for face recognition using GPU

License: Other

CMake 0.48% C 1.18% C++ 95.11% Cuda 3.23%
cuda deep-learning neural-network gpu face-recognition remote-development

gpu-cnn's Introduction

Accelerate CNN computation with GPU

Please indicate the source below when this repo is shared somewhere else:

Source code: https://github.com/WalkerLau/GPU-CNN

The purpose of this repo is to accelerate VIPLFaceNet, a face recognition algorithm with 7 convolution layers, with GPU and CUDA programming.

For more infomation about VIPLFaceNet:

https://github.com/seetaface/SeetaFaceEngine

Development Environment

  • HARDWARE

    • Machine: Amazon AWS EC2, g3s.xlarge (Ohio, us-east-2)

    • GPU: Tesla M60

    • Compute Capability: 5.2

  • SOFTWARE

    • AMI ID: Deep Learning Base AMI (Ubuntu) Version 19.1 (ami-04fde1417a0168c95)

    • Ubuntu 16.04 (contained in AMI)

    • OpenCV 2.4.9.1 (contained in AMI)

    • CUDA 10.0.130 (contained in AMI)

Installation

  1. Download this repo

    git clone https://github.com/WalkerLau/GPU-CNN.git

    Note: The model file GPU-CNN/model/seeta_fr_v1.0.bin is stored in Git Large File Storage (LFS). You should Download and install Git Large File Storage (LFS) before further operations on Linux machine, or download the seeta_fr_v1.0.bin independently from this repo's webpage. Simply "git clone" would not download the complete seeta_fr_v1.0.bin and will cause error. The complete seeta_fr_v1.0.bin should be of about 110 MB.

  2. Build

    cd GPU-CNN
    
    mkdir build
    
    cd build
    
    cmake ../src
    
    make
    

    Note: Try switching to root if build failed.

  3. Run

    ./GPU-CNN

Others

  • About source files:

    • test_face_recognizer.cpp contains top level function

    • conv_net.cpp calls convolution layers

    • inner_product_net.cpp calls fully-connected layers

    • math_functions.cu defines CUDA kernel functions

References

gpu-cnn's People

Contributors

walkerlau avatar

Stargazers

 avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar

Watchers

 avatar  avatar  avatar  avatar

gpu-cnn's Issues

Git clone

您好,我是从西电睿思上面看到一直关注之前FPGA的那个神经网络的加速项目,现在想要下载您的这个GPU的这个项目,但是里面的modul里面的那个大文件一直下载不下来,请问有别的方式下载您的整个工程文件吗?

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.