caffe计算模型中每层参数的个数

import sys
sys.path.insert(0,"D:\CNN\caffe\python")
import caffe


model="MobileFaceNet_deploy.prototxt"


def main():
    net=caffe.Net(model,caffe.TEST)
    total=0
    for item in net.params.items():
        name,layer=item
        print(name+":"+str(layer[0].count))
        total+=layer[0].count
    print("total")
    print(total)
if __name__ == '__main__':
    main()
阅读更多
文章标签: caffe
个人分类: 深度学习
想对作者说点什么? 我来说一句

没有更多推荐了,返回首页

关闭
关闭
关闭