arcface

ArcFace MXNet Symbol=>ONNX=>TensorRT

1.Reference

• arcface github: https://github.com/deepinsight/insightface/tree/master/recognition
• arcface arxiv: ArcFace: Additive Angular Margin Loss for Deep Face Recognition
• get pretrained model: BaiduDrive and Dropbox

2.Export ONNX Model

python3 export_onnx.py --input_shape 4 3 112 112

3.Build arcface_trt Project

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

4.run arcface_trt

cd ../../bin/
./tensorrt_inference arcface ../configs/arcface/config.yaml ../samples/faces_recognition

results:

result
	1	0.51497477	0.83092833	0.44836619	0.44409686	0.44004413	0.57703531	0.48046044	0.50348091	0.52596587
	0.51497477	1	0.5157097	0.49093315	0.48639575	0.55684233	0.41457996	0.4557389	0.45707369	0.51120299
	0.83092833	0.5157097	1	0.43384337	0.43466485	0.44116706	0.55737579	0.49809921	0.50180018	0.52988255
	0.44836619	0.49093315	0.43384337	1	0.8184306	0.52917022	0.44513768	0.51536781	0.50124043	0.56127048
	0.44409686	0.48639575	0.43466485	0.8184306	1	0.53311759	0.48287207	0.50482482	0.52335793	0.49513683
	0.44004413	0.55684233	0.44116706	0.52917022	0.53311759	1	0.46499243	0.51840144	0.4833495	0.43685332
	0.57703531	0.41457996	0.55737579	0.44513768	0.48287207	0.46499243	1	0.53517133	0.51514965	0.48933336
	0.48046044	0.4557389	0.49809921	0.51536781	0.50482482	0.51840144	0.53517133	1	0.4795776	0.47983229
	0.50348091	0.45707369	0.50180018	0.50124043	0.52335793	0.4833495	0.51514965	0.4795776	1	0.51290995
	0.52596587	0.51120299	0.52988255	0.56127048	0.49513683	0.43685332	0.48933336	0.47983229	0.51290995	1