caffe：将mean.binaryproto文件转化为mean.npy文件

swiftfake

于 2018-06-08 15:41:31 发布

阅读量511

点赞数

CC 4.0 BY-SA版权

分类专栏： caffe学习之路文章标签： caffe mean.npy mean mean.binaryproto

本文链接：https://blog.youkuaiyun.com/swiftfake/article/details/80624082

caffe学习之路专栏收录该内容

4 篇文章

订阅专栏

本文介绍了一个简单的Python脚本，该脚本能够将Caffe的mean.binaryproto文件转换为mean.npy文件，方便在深度学习项目中使用。通过解析binaryproto格式的数据并将其转化为NumPy数组进行保存。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

# -*- 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)