Может ли кто-нибудь найти количество нейронов и размер ядра в приведенном ниже коде модели CNN? Я хочу отобразить каждый нейрон внутри слоя на диаграмме

Question

Я хочу знать количество нейронов и размер ядра в приведенном ниже коде модели CNN. Я хочу отобразить каждый нейрон внутри слоя на диаграмме, чтобы визуально проработать мою модель.

Input: Three Values as X,Y, Z    
    //layer:1
        model.add(Conv2D(16, (2, 2), activation = 'relu', input_shape = x_train[0].shape)) 
        model.add(Dropout(0.1))  

        //layer:2 
        model.add(Conv2D(32, (2, 2), activation = 'relu'))   model.add(Dropout(0.2))  model.add(Flatten())

        //layer:3
        model.add(Dense(64, activation= 'relu')) 
        model.add(Dropout(0.5)) 

        //layer:4
        model.add(Dense(3, activation= 'softmax'))

    Output: Three Values as X, Y, Z 

введите описание изображения здесь

Answer 1

Ваш код, по-видимому, использует обозначение, подобное керасу. Если это правильно, то я думаю, что ваши определения слоев следующие:

Conv2D (NUMBER_FILTERS, (SIZE_FILTER), ...)

Плотный (NUMBER_NEURONS, ...)

Так, например, у вас есть 16 масок 2х2 для обучения в первом слое, то есть 64 нейрона.

Чтобы создать визуальное представление вашего CNN, я предлагаю проверить эту ссылку:

http://alexlenail.me/NN-SVG/LeNet.html

Однако, если вы хотите проанализировать сложность вашей ИНС, то поиск числа параметров, которые вам нужно обучить, может быть более представительным, чем число нейронов. В используемой вами библиотеке, вероятно, есть встроенная функция, которая делает это за вас.