search

OpenGVLab / InternImage

[CVPR 2023 Highlight] InternImage: Exploring Large-Scale Vision Foundation Models with Deformable Convolutions
https://arxiv.org/abs/2211.05778
MIT License
2.52k stars 234 forks source link

To trt model have problem! #180

Closed xuweidongkobe closed 1 year ago

xuweidongkobe commented 1 year ago

I have built the trt plugins and load successfully. I use the following code to transfer torch model to trt model, the onnx model has benn transferred successfully, but trt model have problem.

python deploy.py "./deploy/configs/mmdet/instance-seg/instance-seg_tensorrt_dynamic-320x320-1344x1344.py" "./configs/coco/cascade_internimage_xl_fpn_1x_coco.py" "./models/cascade_internimage_xl_fpn_1x_coco.pth" "./deploy/demo.jpg" --work-dir "./trt_model/" --device cuda --dump-info

problem

I tried using different versions of onnx, but it still hasn't been resolved. The trt version is 8.2.3.0. Can you give me some advice？ Thanks.

j0987834204 commented 1 year ago

I upgraded TensorRT version to solve the problem.

xuweidongkobe commented 1 year ago

I upgraded TensorRT version to solve the problem.

hi，can you tell me which trt version you used? @j0987834204

xuweidongkobe commented 1 year ago

I solved the problem with Tensort v8.6.1.6.