foxai.explainer.computer_vision package

Subpackages

Submodules

foxai.explainer.computer_vision.model_utils module

File contains functions to modifiy DNN models.

foxai.explainer.computer_vision.model_utils.get_last_conv_model_layer(model: Module) Module[source]

Get the last convolutional layer from the torch model.

Parameters:

model – torch.nn.Module

Returns:

The last convolutional layer of the model.

Raises:

ValueError if the model does not contain convolutional layers.

foxai.explainer.computer_vision.model_utils.modify_modules(model: Module) Module[source]

Modify modules of given model.

Function iterates over all modules and sets property inplace to False for every torch.nn.ReLU activation function.

Parameters:

model – Neural network object to be modified.

Returns:

Modified neural network object.

Module contents