Caffe网络中具有相同类型名但计算功能不同的层。比如DetectionOutput层,需要使用算子映射指明为FSRDetectionOutput、SSDDetectionOutput等检测算子类型,否则ATC生成离线模型会执行失败。为了避免出现错误,以下两种方案二选一即可。
方案1:可以在ATC命令中加入--op_name_map参数,参考ATC参数中op_name_map参数设置。
方案2:可以在原始网络proto模型文件中将输出算子类型指定为SSDDetectionOutput等算子类型,如下图所示。
图1 输出算子类型修改前(左)和修改后(右)
