文档
注册
评分
提单
论坛
小AI

auto_decomposition

功能说明

根据用户提供的Caffe原始模型(.prototxt和.caffemodel),生成张量分解后的模型文件以及权重文件。

约束说明

  • 用户提供用于转换的原始Caffe模型,需要保证prototxt文件和caffemodel配套。
  • 用户调用张量分解接口函数,输入原始模型路径以及模型分解后的保存路径,接口函数对符合分解条件的卷积层进行自动分解,约束请参见分解约束

函数原型

auto_decomposition(model_file,weights_file,new_model_file,new_weights_file)

参数说明

参数名

输入/返回值

含义

使用限制

model_file

输入

用户Caffe模型的定义文件,格式为.prototxt。

数据类型:string

weights_file

输入

用户训练好的的Caffe模型权重文件,格式为.caffemodel。

数据类型:string

new_model_file

输入

张量分解后的Caffe模型定义文件,格式为.prototxt,例如xx_tensor_decomposition.prototxt

数据类型:string

new_weights_file

输入

张量分解后的Caffe模型权重文件,格式为.caffemodel,例如xx_tensor_decomposition.caffemodel

数据类型:string

返回值说明

无。

函数输出

  • 张量分解后的模型定义文件,格式为.prototxt。
  • 张量分解后的模型权重文件,格式为.caffemodel。

调用示例

from amct_caffe.tensor_decompose import auto_decomposition
auto_decomposition(model_file='ResNet-50-deploy.prototxt',
                  weights_file='ResNet-50-weights.caffemodel',
                  new_model_file='ResNet-50-deploy_tensor_decomposition.prototxt',
                  new_weights_file='ResNet-50-deploy_tensor_decomposition.caffemodel')
搜索结果
找到“0”个结果

当前产品无相关内容

未找到相关内容,请尝试其他搜索词