Mask RCNN on JetPack 6.1 on AGX Orin

Robotics & Edge Computing Jetson Systems Jetson AGX Orin
1

Hi, is anyone able to confirm for me that I able able to run Mask RCNN while using PyTorch on JetPack 6.1? I have been able to use PyTorch to custom train classification models, then I convert the model using trtexec.

I am thinking forward to my future project. I know PyTorch has Mask RCNN available, will I be able to custom train and deploy a Mask RCNN model with PyTorch and TensorRT? I am not sure if there’s more to segmentation models versus the classification (specifically ResNet) models I have been working with.

Thanks,

Andrew

5

Hi,

Yes. Pytorch and TRT are supported.

We have a TensorFlow tutorial available. Please refer to the following link:
https://github.com/NVIDIA/TensorRT/tree/main/samples/python/tensorflow_object_detection_api

Thanks

7

This topic was automatically closed 14 days after the last reply. New replies are no longer allowed.