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

本文介绍了一个简单的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)

如果出现黑屏应该是正常情况,关闭程序即可,然后就可以在预先设定的文件夹内找到生成的mean.npy文件

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值