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()
阅读更多

扫码向博主提问

迷若烟雨

领先的深度学习解决问题提供商
  • 擅长领域:
  • 深度学习
  • 人脸识别
去开通我的Chat快问
想对作者说点什么?

博主推荐

换一批

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