扩展算子列表

过程算子

Reverse

将输入tensor指定的轴进行反转。

Upsample

使用pooling mask的上采样。

用于yolo网络。

Normallize

将SSD网络中Channel维度中的元素在L2进行归一化。

Reorg

Reorg layer in Darknet，将通道数据转移到平面上，或反过来操作。

对应算子规格中的PassThrough算子。

ROIAlign

从feature map中获取ROI（range of interest）的特征矩阵。

ShuffleChannel

把channel维度分为[group, channel/Group]，再在channel维度中进行数据转置[channel/Group,group]。

Yolo（Yolo/Detection/Region）

Yolo/Detection/Region算子，都需要替换为Yolo算子，对卷积网络输出的feature map生成检测框的坐标信息，置信度信息及类别概率。

PriorBox

用于SSD网络，根据输入参数，生成prior box。

SpatialTransformer

用于仿射变换。