SSD MobileNetV2 TensorRT incompability

dusty-nv / pytorch-detection

Training of object detection networks with PyTorch

16 stars 22 forks source link

SSD MobileNetV2 TensorRT incompability #2

Open tmralmeida opened 4 years ago

tmralmeida commented 4 years ago

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!

donipologhimire commented 2 years ago

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.