Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Are TFLite models supported for RISE Perturbation + DRISE #123

Open
ai2ys opened this issue Aug 19, 2022 · 1 comment
Open

Are TFLite models supported for RISE Perturbation + DRISE #123

ai2ys opened this issue Aug 19, 2022 · 1 comment

Comments

@ai2ys
Copy link

ai2ys commented Aug 19, 2022

I would like to try "RISE Perturbation + DRISE" similar to the Notebook https://github.com/XAITK/xaitk-saliency/blob/master/examples/DRISE.ipynb from the examples, but with a TFLite model.

In the video on the website https://www.kitware.com/kitware-receives-honorable-mention-for-explainable-ai-toolkit/ is gets mentioned that TensorFlow models are supported by XAITK-saliency, but I can not find any information on this.

@Purg
Copy link
Member

Purg commented Aug 19, 2022

Hi! TF models are supported notionally in that a DetectImageObjects implementation could be written to wrap the use of something from TensorFlow (our adapter interface does not use any framework-specific primitives). We do not have any stock implementations that utilize TensorFlow that would be turn-key right now. Some of our notebooks examples (specifically this one) show the creation of an implementation of this interface. The referenced example still uses pytorch, but I hope the mechanism makes sense enough that you could adapt it for your TensorFlow use case.

You may have already deduced the following, but I'll mention it anyway just in case:

  • The model modularity in the DRISE example notebook happens in the blackbox_detector slot in the sal_map_generator.generate(ref_image, ref_bboxes, ref_scores, blackbox_detector) call.
  • We define object-detection-oriented interface (here) that we expect the input to be an implementation of the DetectImageObjects, which is an adapter pattern interface defined in the smqtk_detection package (specific interface here).
  • If you create a new implementation for DetectImageObjects that you think might be generally useful, the smqtk-detection package will happily accept contributions! (I/we also maintain that.)

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

No branches or pull requests

2 participants