README.md

CenterFace Network ONNX=>TensorRT

1.Reference

• CenterFace github: https://github.com/Star-Clouds/CenterFace
• CenterFace arxiv: CenterFace: Joint Face Detection and Alignment Using Face as Point
• get pretrained model: centerface.onnx or centerface_bnmerged.onnx

2.Export ONNX Model

• export centerface.onnx model

python3 export_onnx.py

• export centerface_bnmerged.onnx model

python3 export_onnx.py --pretrained ./centerface_bnmerged.onnx --onnx ./centerface_bnmerged_transpose.onnx

3.Build tensorrt_inference Project

cd ../  # in project directory
mkdir build && cd build
cmake ..
make -j

4.Run tensorrt_inference

cd ../../bin/
./tensorrt_inference CenterFace ../configs/CenterFace/config.yaml ../samples/faces_detection

5.Results