DeepStream-Yolo icon indicating copy to clipboard operation
DeepStream-Yolo copied to clipboard

custom yolov7 model works in tensorrt, but not in deepstream. > deserializeEngineAndBackend()

Open HeungJunKim opened this issue 2 years ago • 3 comments

nvinfer gstnvinfer.cpp:638:gst_nvinfer_logger:<primary_gie> NvDsInferContext[UID 1]: Info from NvDsInferContextImpl::deserializeEngineAndBackend() <nvdsinfer_context_impl.cpp:1912> [UID = 1]: deserialized trt engine from :/home/nvidia/workspace/deepstream_sdk/DeepStream-Yolo/yolov7x.engine

deepstream-app error >> deserializeEngineAndBackend()

It looks similar to the error in tensorrt

[07/28/2022-15:02:55] [TRT] [E] 1: [pluginV2Runner.cpp::load::290] Error Code 1: Serialization (Serialization assertion creator failed.Cannot deserialize plugin since corresponding IPluginCreator not found in Plugin Registry) [07/28/2022-15:02:55] [TRT] [E] 4: [runtime.cpp::deserializeCudaEngine::50] Error Code 4: Internal Error (Engine deserialization failed.)

I solved the problem by adding the following function in tensorrt initLibNvInferPlugins(&gLogger, "");

HeungJunKim avatar Aug 09 '22 23:08 HeungJunKim

There's no support for YOLOv7 yet.

marcoslucianops avatar Aug 09 '22 23:08 marcoslucianops

아직 YOLOv7에 대한 지원이 없습니다.

Waiting for yolov7 support. Thanks for your reply.

HeungJunKim avatar Aug 10 '22 04:08 HeungJunKim

Added support for YOLOv7.

marcoslucianops avatar Aug 12 '22 19:08 marcoslucianops