albermax / innvestigate

A toolbox to iNNvestigate neural networks' predictions!
Other
1.25k stars 233 forks source link

The execution of this package is very slow #129

Closed Tahedi1 closed 5 years ago

Tahedi1 commented 5 years ago

Is there any batch version for this package. Can I use the analyser for a batch of images ?

albermax commented 5 years ago

Hi Mohammed,

sorry to hear. Could you please a bit more specific? Are you using a GPU? Yes, you have to use a batch of images, i.e., the first axis of the input is always the batch axis.

Cheers, Max

Tahedi1 commented 5 years ago

Hi Albert,

Thanks for giving me the information about the use of batch. I'm using GeForce GTX 1080 Ti GPU. The analysis of one image (batch) take more than 2 second.

Kind regards

albermax commented 5 years ago

At the first analysis the graph is compiled. Every further analysis should be faster.

Cheers, Max

Tahedi1 commented 5 years ago

Thanks albermax.

albermax commented 5 years ago

Let us know if you have any further troubles!