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

3.TRT

INPUT

[batch_size,3,640,640]

OUTPUT

[batch_size,1,160,160]

[batch_size,2,160,160]

[batch_size,2,160,160]

[batch_size,10,160,160]

dynamic shape

define the input/output data as variables of dynamic shape (batch_size,3,640,640).

issue

4.Results