1. 程式人生 > >caffe:將mean.binaryproto檔案轉化為mean.npy檔案

caffe:將mean.binaryproto檔案轉化為mean.npy檔案

# -*- 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檔案