卷积层与全连接层权重(参数)个数的计算

本文介绍了在深度学习环境中,特别是使用TensorFlow 2.0.0b0构建的卷积神经网络(CNN)中,卷积层和全连接层的权重个数计算方法。卷积层的参数数量由卷积核尺寸、通道数和卷积核个数决定,而全连接层的参数数量取决于上一层神经元的数量和当前层的神经元数量。文中给出了具体的计算示例,如卷积层1的参数为320,全连接层2的参数为650。
摘要由CSDN通过智能技术生成

环境:Pycharm2017.2,tensorflow 2.0.0b0,Win7,python3.6

感想:最近在研究卷积神经网络的构建和训练,权重(参数)是一个非常重要的概念,卷积层会用多个不同的卷积核,每个卷积核有m*n个权重,用来提取和强化图像的特征,主要利用了图像的局部相关性。全连接层,顾名思义,每一层的每个神经元都与上一层的每个神经元连接起来,每条连接都有一个权重和偏置来进行传递。

构建卷积网络如下:

import tensorflow as tf
from tensorflow.python.keras import datasets, models, layers
import os
class CNN(object):
    def __init__(self):
        model = models.Sequential()
        # 第1层卷积,卷积核大小为3*3,32个卷积核,28*28为待训练图片的大小
        model.add(layers.Conv2D(32, (3, 3), activation='relu', input_shape=(28, 28, 1)))
        # 池化层
        model.add(layers.MaxPooling2D(2, 2))
        # 第2层卷积,卷积核大小为3*3,64个卷积核
        model.add(layers.Conv2D(64, (3, 3), activation='relu'))
        # 池化层
        model.add(layers.MaxPooling2D(2, 2))
        # 第3层卷积,卷积核大小为3*3,128个卷积核
        mod
评论 2
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值