alexander-pv / maskrcnn_tf2

Mask R-CNN for object detection and instance segmentation with Keras and TensorFlow V2 and ONNX and TensorRT optimization support.
Other
40 stars 11 forks source link

TensorRT inference fails in TensorRT8 #24

Open sallydly opened 1 year ago

sallydly commented 1 year ago

Hello, I've been following the example listed out in the README to create a basic MaskRCNN model using the balloon dataset I'm currently using TensorRT 8.4.3.1 and am getting this error when I run inference on the model (it is building into an engine perfectly fine): 2: [pluginV2Runner.cpp::execute::265] Error Code 2: Internal Error (Assertion status == kSTATUS_SUCCESS failed. ) I've tested a similar model in TensorRT 7.2 and it is able to run inference with no error. Any insight on what the issue could be between the updates?

I cannot use TensorRT 7 since I am using Ubuntu20.04 which only has packages for TensorRT 8+.