Caffeラーニング:pycaffeを使用してmeanを生成するfile.pyファイル
build/tools/computeの利用image_meanは
mean.binaryproto
ファイルを生成した後、以下のコードを利用してmean.npy
ファイルを生成し、pycaffeに呼び出す.#!/usr/bin/env python
MEAN_BIN = 'mean.binaryproto'
MEAN_NPY = 'mean.npy'
from caffe.proto import caffe_pb2
from caffe.io import blobproto_to_array
print('generating mean file...')
mean_blob = caffe_pb2.BlobProto()
mean_blob.ParseFromString(open(MEAN_BIN, 'rb').read())
import numpy as np
mean_npy = blobproto_to_array(mean_blob)
mean_npy_shape = mean_npy.shape
mean_npy = mean_npy.reshape(mean_npy_shape[1], mean_npy_shape[2], mean_npy_shape[3])
np.save(file(MEAN_NPY, 'wb'), mean_npy)
print('done...')