pcb9382 / FaceAlgorithm

face detection face recognition包含人脸检测(retinaface,yolov5face,yolov7face,yolov8face),人脸检测跟踪(ByteTracker),人脸角度计算(Face_Angle)人脸矫正(Face_Aligner),人脸识别(Arcface),口罩检测(MaskRecognitiion),年龄性别检测(Gender_age),静默活体检测(Silent_Face_Anti_Spoofing),FaceAlignment(106keypoints)
MIT License
284 stars 60 forks source link

Onnx2Trt Conversion Problem #6

Open SwEngine opened 9 months ago

SwEngine commented 9 months ago

When I tried to convert my onnx file to trt file (tensorrt7), I got the error below, how can I solve it?

[12/11/2023-16:10:27] [I] [TRT] ModelImporter.cpp:135: No importer registered for op: ScatterND. Attempting to import as plugin. [12/11/2023-16:10:27] [I] [TRT] builtin_op_importers.cpp:3659: Searching for plugin: ScatterND, plugin_version: 1, plugin_namespace: [12/11/2023-16:10:27] [E] [TRT] INVALID_ARGUMENT: getPluginCreator could not find plugin ScatterND version 1 ERROR: builtin_op_importers.cpp:3661 In function importFallbackPluginImporter: [8] Assertion failed: creator && "Plugin not found, are the plugin name, version, and namespace correct?" [12/11/2023-16:10:27] [E] Failed to parse onnx file [12/11/2023-16:10:27] [E] Parsing model failed [12/11/2023-16:10:27] [E] Engine creation failed [12/11/2023-16:10:27] [E] Engine set up failed

pcb9382 commented 8 months ago

Please use TRT8, trt7 may need to be changed by yourself due to some operator unsupport and interface problems