TensorRt models

[Hasnain1997-ai]

Hi there!

thank you for the wonderful work you have done

I wanted to know that have you worked on converting your models to TensorRT

I mean segmentation and classification models?

can you help me in this regards

Thanks

[fishial]

I'll pass on your request!

[AndrzejCodahead]

@Hasnain1997-ai, we haven't converted the models to TensorRT. For segmentation model you can try follow by this notebook, for the classification model, this is a slightly modified resnet18 you can try this repo