资源简介

包括imagenet_mean.binaryproto,由imagenet_mean.binaryproto转为npy格式的mean.npy,还包括两种格式相互转换的python代码;包含imagenet标签/train/test/val文件及种类文件

资源截图

代码片段和文件信息

import numpy as np
import caffe
import sys

blob = caffe.proto.caffe_pb2.BlobProto()
data = open( ‘imagenet_mean.binaryproto‘  ‘rb‘ ).read()
blob.ParseFromString(data)
arr = np.array( caffe.io.blobproto_to_array(blob) )
out = arr[0]
np.save( ‘mean.npy‘  out )

评论

共有 条评论