Working with Yolo-Segmentation models

stavrostheocharis / easy_explain

An XAI library that helps to explain AI models in a really quick & easy way

MIT License

12 stars 1 forks source link

Working with Yolo-Segmentation models #15

Open krishnaagastya opened 3 weeks ago

krishnaagastya commented 3 weeks ago

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 commented 2 weeks ago

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 commented 4 days ago

Hi,

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

Thanks in advance.

Rgds