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()

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

私密
私密原因:
请选择设置私密原因
  • 广告
  • 抄袭
  • 版权
  • 政治
  • 色情
  • 无意义
  • 其他
其他原因:
120
出错啦
系统繁忙,请稍后再试

关闭