计算FLOPs和Parameters数量

在卷积中要计算的parameter数量:

输入:                【H_in, W_in, C_in】

卷积核大小:      【K_h, K_w】

输出:                【H_out, W_out, C_out】

 

parameters = (K_h * K_w * C_in)*C_out + C_out(偏置)

 

FLOPs是指: floating point operations: 代表浮点运算次数

计算FLOPs和Parameters数量_第1张图片

 

FLOPs = parameters * (H_out * W_out) = ((K_h * K_w * C_in)*C_out + C_out) * (H_out * W_out)

 

计算卷积层后的图像尺寸

输入图像尺寸:W*W

卷积核尺寸: K*K

stride:  S

padding: P

 

输出尺寸: output_size = (W - K + 2P) / S + 1

 

池化操作后的图像尺寸:

输入图像尺寸:W*W

卷积核尺寸: K*K

stride:    S

 

池化后的size: output_size = (W - K) / S + 1

你可能感兴趣的:(深度学习,机器学习)