pytorch-detection icon indicating copy to clipboard operation
pytorch-detection copied to clipboard

Published 20 hours ago • verified publisher icon dusty-nv

SSD MobileNetV2 TensorRT incompability

Open tmralmeida opened this issue 4 years ago • 1 comments

Hi @dusty-nv ,

I tried to train a custom SSD w/ mibilenetV2 from this repo and then deploy on a jetson agx xavier through jetson-inference. However, it seems that there is an incompability between onnx and tensorrt. I obtained this error:

error

My tensorrt version is 5.1.6.

Could you help me on that? Thanks!

tmralmeida avatar May 26 '20 09:05 tmralmeida

Hi @tmralmeida , can you help how you custom trained the model.

  • can you guide me how the dataset should look like, I have images (.jpeg) and label (.xml files).
  • I see that the train_ssd.py, is used for training, however how do I load a pre trained model, if I have none. For example, detecting rocks.

donipologhimire avatar Sep 15 '22 01:09 donipologhimire