In the method "visualize_convolutions" before plotting the output of each layer of the model, some processing has been performed as below - features -