Skip to content
master
Switch branches/tags
Code

Latest commit

 

Git stats

Files

Permalink
Failed to load latest commit information.
Type
Name
Latest commit message
Commit time
 
 
 
 
 
 
 
 
 
 
 
 

Face-Recognition

This sample targets for demonstrating TensorRT2.1 Plugin API

We leverage most of the functions from jetson_inference; please check it first if you need more DL samples:

https://github.com/dusty-nv/jetson-inference



Environment

Jetson TX2
JetPack-3.1


Build

$ sudo apt-get install git cmake
$ git clone https://github.com/AastaNV/Face-Recognition
$ cd Face-Recognition
$ mkdir build
$ cd build
$ cmake ..
$ make

Run

$ cd aarch64/bin
$ ./face-recognition

Plugin Layer

BboxMergeLayer
This plugin layer demonstrate how to implement a CPU-based Plugin layer

  1. Make required tensor as output
  2. Allocate unified memory: CPU pointer== GPU pointer

DataRoiLayer
This plugin layer demonstrate how to implement a GPU Plugin layer

  1. Got input/output data pointer in enqueue function
  2. Launch GPU kernel with same cuda stream

RecognitionLayer
This plugin layer demonstrate more complicated handling of Plugin layer

  1. This class can handle two differient layers: selectBbox and summaryLabel
  2. Define some shared variable to make between layers communication easier


Support

Please rise your problem in our forum to get immediately support.
https://devtalk.nvidia.com/default/board/189/jetson-tx2/


About

Demonstrate Plugin API for TensorRT2.1

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published