Working with Yolo-Segmentation models

[krishnaagastya]

Hi,

This maybe more in lines of an enhancement request. Do let me know.

The Easy explainer works well with Object detection and bounding boxes. It throws the below exception when I use a Segmentation model.

• NotImplementedError: Relevance propagation not implemented for layer type <class 'ultralytics.nn.modules.head.Segment'>

Is there a plan to include this in the package?

Do share your roadmap if possible.

Rgds
Krishna

[stavrostheocharis]

Hello, to tell you the truth, I have not a specific roadmap for this lib, since I work alone on it, on some spare time I may have. If you want, you are free to help with the segmentation xai part and try to include it in a new version. I will have it in my mind to work a bit also on it and try to create it. Thanks!

[krishnaagastya]

Hi,

Do you have any documentation on the flow. Can you share anything you have prepared?

Thanks in advance.

Rgds