只有caffemodel文件,反推prototxt

来源:互联网 发布:俄罗斯离婚率 知乎 编辑:程序博客网 时间:2024/06/16 01:45

参考只有caffemodel文件可以反推prototxt吗?.

from caffe import caffe_pb2def toPrototxt(modelName, deployName):    with open(modelName, 'rb') as f:        caffemodel = caffe_pb2.NetParameter()        caffemodel.ParseFromString(f.read())    for item in caffemodel.layers:        item.ClearField('blobs')    for item in caffemodel.layer:        item.CLearField('blobs')    with open(deployNmae, 'w') as f:        f.write(str(caffemodel))if __name__ == "__main__":    modeName = 'test.caffemodel'    deployName = 'test.prototxt'    toPrototxt(modelName, deployName) 
原创粉丝点击