Skip to content

Latest commit

 

History

History
 
 

CenterFace

CenterFace Network ONNX=>TensorRT

1.Reference

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.onnx

3.Build CenterFace_trt Project

mkdir build && cd build
cmake ..
make -j

4.Run CenterFace_trt

./CenterFace_trt ../config.yaml ../samples

5.Results