caffe:將mean.binaryproto檔案轉化為mean.npy檔案
阿新 • • 發佈:2019-02-15
# -*- coding: utf-8 -*-
"""
Created on Thu May 24 16:03:14 2018
@author: chrisd
"""
import caffe
import numpy as np
mean_proto_path='×××/mean.binaryproto' #存放mean.binaryproto檔案的路徑
mean_npy_path='×××/mean.npy' #用於存放生成的mean.npy檔案的路徑
blob =caffe.proto.caffe_pb2.BlobProto()
data =open(mean_proto_path,'rb' ).read()
blob.ParseFromString(data)
array =np.array(caffe.io.blobproto_to_array(blob))
mean_npy=array[0]
np.save(mean_npy_path,mean_npy)
如果出現黑屏應該是正常情況,關閉程式即可,然後就可以在預先設定的資料夾內找到生成的mean.npy檔案