卷积神经网络(CNN)的计算量和参数量计算

今天看文章复习了一下计算量和参数量的计算方法,这里做一下笔记

忽略padding和stride的话,输入层,卷积核和输出层的关系,以及参数量、计算量如下(第二行是一个例子)

卷积神经网络(CNN)的计算量和参数量计算_第1张图片

关于参数量:每个卷积核有 K × K × M K×K×M K×K×M个参数,那么共有N个卷积核,参数量就是 K × K × M × N K×K×M×N K×K×M×N

关于计算量:每个卷积核遍历输入层,计算量是 K × K × M × W o × W o K×K×M×W_o×W_o K×K×M×Wo×Wo,那么N个卷积核的计算量就再乘上N。(这里乘上两个 W o W_o Wo我没有做严格的证明)

你可能感兴趣的:(神经网络,人工智能,卷积,深度学习,卷积神经网络)