The purpose of this project is to apply mediapipe to more AI chips.
Our Official Website: www.houmo.ai
Who We Are: We are Houmo - A Great AI Company.
We wish to change the world with unlimited computing power,
We will subvert the AI chip with in memory computing.
This Project we created is the first one that migrate tensorrt inference engine into Google Mediapipe.
The purpose of this project is to apply mediapipe to more AI chips.
a)gcc and g++ version 8.4.0 (Ubuntu/Linaro 8.4.0-1ubuntu1~18.04)
install command:
$sudo apt install gcc-8 g++-8
$sudo update-alternatives --install /usr/bin/gcc gcc /usr/bin/gcc-8 800 --slave /usr/bin/g++ g++ /usr/bin/g++-8
check gcc g++ version
$gcc -v
$g++ -v
b)Mediapipe Dependencies
Please refer to Mediapipe Installation and install all required software packages and runtime libraries.
c)Cuda Related
We created two soft links pointing to cuda headers. Change to your own paths if needed.
**/mediapipe_plus/third_party/cuda_trt$ tree
.
├── BUILD
├── usr_local_cuda
│ └── include -> /usr/local/cuda/include
└── usr_local_cuda-10.2
└── include -> /usr/local/cuda-10.2/targets/aarch64-linux/include/
Please follow the instructions below to compile and run our demo.
Warning:Before Upgrading, Please BackUp Your AGX Xavier To Prevent Data Loss
Refer to xavier official website:Over-the-Air Update:
Section: Updating a Jetson Device -> To update to a new minor release
Clone
$git clone ***
$cd **
Build the Demo
$bazel build //calculators/tensorrt:trt_inference_calculator_test
Run
GLOG_logtostderr=1 ./bazel-bin/calculators/tensorrt/trt_inference_calculator_test --input_video_path=./test1.mp4 --remote_run=true
Expected Output
Under current folder, there will be a video file named "trt_infer.mp4" be generated.
Each frame has detected facial boxes and facial points.Like This:
We created several calculators under directory "./calculators/" to build a TensorRT Engine from onnx .
And the target:trt_inference_calculator_test is a face detection demo to show how to use these calculators.
Face Detection Demo is an ultrafast face detection solution that comes with 6 landmarks and multi-face support.
It is based on BlazeFace, a lightweight and well-performing face detector.
We left several TODOs which will be done in next version .