AastaNV / TRT_object_detection

Python sample for referencing object detection model with TensorRT
142 stars 27 forks source link

TensorRT reshape volume mismatch #3

Open cosama opened 5 years ago

cosama commented 5 years ago

I tried to use some other models not included in the list and run into an issue with the reshape operator producing an Volume Mismatch error.

Looking at this thread this seems to be a known issue with TensorRT not supporting this operator at this point. Is there a way to go around this with a plugging of some sort? Which models will you be supporting in the near future?

Thanks.