jkjung-avt / tensorrt_demos

TensorRT MODNet, YOLOv4, YOLOv3, SSD, MTCNN, and GoogLeNet
https://jkjung-avt.github.io/
MIT License
1.74k stars 545 forks source link

Convert ssd_mobilenet_v2_coco to TensorRT #568

Closed kanakedge closed 2 years ago

kanakedge commented 2 years ago

I have been trying to convert ssd_mobilenet_v2_coco to tensorRT. I changed the script at some locations to make it work for ssd_mobilenet_v1_coco, it worked. Now I am looking for ssd_mobilenet_v2_coco but it is giving assertionnmsPlugin.cpp,246.

`TensorRT] VERBOSE: Debug synchronize completed successfully after build for layer: MultipleGridAnchorGenerator copy (type=Reformat, tactic=0) [TensorRT] VERBOSE: Debug synchronize completed successfully after build for layer: MultipleGridAnchorGenerator_1 copy (type=Reformat, tactic=0) [TensorRT] VERBOSE: Debug synchronize completed successfully after build for layer: MultipleGridAnchorGenerator_2 copy (type=Reformat, tactic=0) [TensorRT] VERBOSE: Debug synchronize completed successfully after build for layer: MultipleGridAnchorGenerator_3 copy (type=Reformat, tactic=0) [TensorRT] VERBOSE: Debug synchronize completed successfully after build for layer: MultipleGridAnchorGenerator_4 copy (type=Reformat, tactic=0) [TensorRT] VERBOSE: Debug synchronize completed successfully after build for layer: MultipleGridAnchorGenerator_5 copy (type=Reformat, tactic=1002)

assertionnmsPlugin.cpp,246

Aborted (core dumped)`

Any ideas why this error is coming.

jkjung-avt commented 2 years ago

Same issue as this one: https://github.com/jkjung-avt/tensorrt_demos/issues/553#issuecomment-1161620781

#assertion/opt/tensorrt/TensorRT/plugin/nmsPlugin/nmsPlugin.cpp,246

kanakedge commented 2 years ago

Yes