相关疑难解决方法(0)

如何计算卷积神经网络的参数个数?

我无法提供正确数量的AlexNetVGG Net参数.

例如,要计算conv3-256一层VGG Net 的参数个数,答案是0.59M =(3*3)*(256*256),即(内核大小)*(两者中的两个通道数的乘积)联合层),但是这样,我无法得到138M参数.

那么请你告诉我计算的错误,或者告诉我正确的计算程序?

machine-learning computer-vision neural-network vgg-net

40
推荐指数
3
解决办法
6万
查看次数

如何计算卷积神经网络中的参数总数?

如何计算CNN网络中的参数总数

这是代码:

input_shape = (32, 32, 1)
flat_input_size = input_shape[0]*input_shape[1]*input_shape[2]
num_classes = 4

cnn_model = Sequential()
cnn_model.add(Conv2D(32, (3, 3), padding='same',
                 input_shape=input_shape))
cnn_model.add(Activation('relu'))
cnn_model.add(MaxPooling2D(pool_size=(2, 2)))
cnn_model.add(Conv2D(64, (3, 3)))
cnn_model.add(Activation('relu'))
cnn_model.add(MaxPooling2D(pool_size=(2, 2)))
cnn_model.add(Dropout(0.25))
cnn_model.add(Conv2D(128, (3, 3), padding='same'))
cnn_model.add(Activation('relu'))
cnn_model.add(MaxPooling2D(pool_size=(2, 2)))
cnn_model.add(Dropout(0.25))
cnn_model.add(Flatten())
cnn_model.add(Dense(512))
cnn_model.add(Activation('relu'))
cnn_model.add(Dropout(0.5))
cnn_model.add(Dense(num_classes))
cnn_model.add(Activation('softmax'))
Run Code Online (Sandbox Code Playgroud)

这是结果 在此处输入图片说明

如何获得 320, 18496, 73856, 590336, 2052,谁能解释一下?

machine-learning neural-network deep-learning convolutional-neural-network

1
推荐指数
1
解决办法
2967
查看次数