caffe.Classifier

版权声明:本文为博主原创文章,未经博主允许不得转载。 https://blog.csdn.net/u011961856/article/details/78964404
 gender_net = caffe.Classifier(network, pretrained_model,
                                  channel_swap=(2, 1, 0),
                                  raw_scale=255,
                                  image_dims=(size, size),mean=np.array([104,117,123]))

channel_swap为将图像由RGB转换为BGR.

mean,通常神经网络希望输入均值为0,然而图像的均值为通常大于0,因此要对图像像素值减去一个均值,使得图像均值为0.mean与训练的数据均值设置参数相同.

raw_scale,如果网络输入值范围为0-255,则raw_scale=255,如果输入值范围为0-1,则raw_scale=1

image_dims为网络的输入图像大小.

猜你喜欢

转载自blog.csdn.net/u011961856/article/details/78964404