Future plans for instance segmentation model in tao toolkit

I have a camera in a fitted environment where I need:

• An edge-friendly model
• Instance segmentation output

At GTC last year, I was informed that a new implementation of Mask R-CNN would be added to TAO Toolkit and that class_weight would be introduced in the configuration. Now, a year later, I have the following questions:

1. Are there any plans to add a new implementation of Mask R-CNN or a more edge-friendly instance segmentation model to TAO Toolkit (e.g., YOLOv8-Seg)?
2. Would it be possible to include class_weight/weighted loss functions in the configuration?
3. Are there any plans to introduce an instance segmentation solution for BYOM (Bring Your Own Model)?

2 posts - 2 participants

Read full topic