search

deepinsight / insightface

State-of-the-art 2D and 3D Face Analysis Project
https://insightface.ai
22.18k stars 5.28k forks source link

arcface onnx #1274

Open alicera opened 3 years ago

alicera commented 3 years ago

model : r50-arcface command: python convert_onnx.py

mxnet version: 1.5.0 onnx version: 1.2.1 input-shape: (1, 3, 112, 112)

THEN DO

command: onnx2trt r50.onnx -o test.trt

Input filename: r50.onnx ONNX IR version: 0.0.3 Opset version: 7 Producer name:
Producer version: Domain:
Model version: 0 Doc string:

Parsing model While parsing node number 5 [PRelu -> "relu0"]: ERROR: /opt/onnx-tensorrt/ModelImporter.cpp:143 In function importNode: [8] No importer registered for op: PRelu

SthPhoenix commented 3 years ago

You can check this script as reference

eeric commented 2 years ago

recognition onnx inference example for single image: https://github.com/eeric/Face_recognition_cnn/tree/main/code/insightface/onnx_inference