Converting FCN8-ResNet18 from Pytorch to TensorRT for inference on Jetson Nano

AI & Data Science Deep Learning (Training & Inference) TensorRT
, , , ,
2

Hi @camille,

Please refer to below link for dynamic shape model:

https://github.com/NVIDIA/TensorRT/tree/master/samples/opensource/sampleDynamicReshape

You can also try trtexec command: (Need to use explicitBatch flag as well)
https://github.com/NVIDIA/TensorRT/tree/master/samples/opensource/trtexec#example-4-running-an-onnx-model-with-full-dimensions-and-dynamic-shapes

Thanks