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

简介

本章节修改只适用于Caffe网络模型。

网络的算子可以分为如下几类:

  • 标准算子:昇腾AI处理器支持的Caffe标准算子,比如Convolution等。
  • 扩展算子:昇腾AI处理器支持的公开但非Caffe标准算子,分为 2 种:
    • 一种是基于Caffe框架进行自定义扩展的算子,比如Faster RCNN中的ROIPooling、SSD中的归一化算子Normalize等。
    • 另外一种是来源于其他深度学习框架的自定义算子,比如YOLOv2中Passthrough等。

Faster RCNN、SSD等网络模型都包含了一些原始Caffe框架中没有定义的算子结构,如ROIPooling、Normalize、PSROI Pooling和Upsample等。为了使昇腾AI处理器能支持这些网络,需要对原始的Caffe框架网络模型进行扩展,降低开发者开发自定义算子/开发后处理代码的工作量。若开发者的Caffe框架网络模型中使用了这些扩展算子,在进行模型转换时,需要先在prototxt中修改/添加扩展层的定义,才能成功进行模型转换。

本章节提供了昇腾AI处理器的扩展算子列表,并给出了如何根据扩展算子修改prototxt文件方法。

搜索结果
找到“0”个结果

当前产品无相关内容

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