Caffe学习

2024-10-12 05:59:10

将binaryproto转为npy

import caffe % suppose caffe is already in the path of Python

import numpy as np

import sys

blob = caffe.proto.caffe_pb2.BlobProto()

data = open( 'data/ilsvrc12/imagenet_mean.binaryproto' , 'rb' ).read()

blob.ParseFromString(data)

arr = np.array( caffe.io.blobproto_to_array(blob) )

arr.shape % check the shape of arr

num_group理解示意图：

最新文章

热门文章