showing pictures of every layers in cnn in matlab

main.m
% Load the trained CNN model
net = alexnet; 

% Get the names of all the layers in the network
layers = net.Layers; 

% Loop through each layer and display the feature maps
for i = 1:numel(layers)
    layer = layers(i);
    
    % Check if the layer is a convolutional layer
    if isa(layer,'nnet.cnn.layer.Convolution2DLayer')
        
        % Compute the activations of the current layer
        inputSize = net.Layers(1).InputSize;
        I = randn(inputSize);
        featureMap = activations(net, I, layer.Name);
        
        % Visualize the feature map using montage
        figure;
        montage(featureMap)
        title(layer.Name)
        
        % Alternatively, visualize the feature map using imshow
        % figure;
        % for j = 1:size(featureMap, 3)
        %     subplot(8, 8, j);
        %     imshow(featureMap(:,:,j));
        % end
        % title(layer.Name)
    end
end
909 chars
33 lines