扩展算子列表【免费下载链接】geGEGraph Engine是面向昇腾的图编译器和执行器提供了计算图优化、多流并行、内存复用和模型下沉等技术手段加速模型执行效率减少模型内存占用。 GE 提供对 PyTorch、TensorFlow 前端的友好接入能力并同时支持 onnx、pb 等主流模型格式的解析与编译。项目地址: https://gitcode.com/cann/ge过程算子算子类型说明Proposal用于Faster R-CNN根据rpn_cls_prob的foregroundrpn_bbox_pred中的bounding box regression修正anchors获得精确的proposals。ROIPooling用于Faster R-CNN中对ROIRegion of interest进行池化操作主要用于目标检测任务。PSROIPooling用于R-FCN中进行位置敏感的候选区域池化操作主要用于目标检测任务。Reverse将输入tensor指定的轴进行反转。Upsample使用pooling mask的上采样。用于yolo网络。Normalize将SSD网络中Channel维度中的元素在L2进行归一化。ReorgReorg layer in Darknet将通道数据转移到平面上或反过来操作。对应算子规格中的PassThrough算子。ROIAlign从feature map中获取ROIRegion of interest的特征矩阵。ShuffleChannel把channel维度分为[group, channel/Group]再在channel维度中进行数据转置[channel/Group,group]。YoloYolo/Detection/Region算子都需要替换为Yolo算子对卷积网络输出的feature map生成检测框的坐标信息置信度信息及类别概率。PriorBox用于SSD网络根据输入参数生成prior box。SpatialTransformer用于仿射变换。后处理算子算子类型说明YoloV3DetectionOutput此算子用于YOLOv3的后处理过程对卷积网络输出的feature map生成检测框的坐标信息置信度信息及类别概率。YoloV2DetectionOutput此算子用于YOLOv2的后处理过程对卷积网络输出的feature map生成检测框的坐标信息置信度信息及类别概率。SSDDetectionOutput此算子用于SSD网络的后处理过程用于整合预选框、预选框偏移以及得分三项结果最终输出满足条件的目标检测框、目标的label和得分。FSRDetectionOutput此算子用于Faster R-CNN网络的后处理过程对结果进行分类并对每个类输出最终的bbox数量、坐标、类别概率及类别索引。【免费下载链接】geGEGraph Engine是面向昇腾的图编译器和执行器提供了计算图优化、多流并行、内存复用和模型下沉等技术手段加速模型执行效率减少模型内存占用。 GE 提供对 PyTorch、TensorFlow 前端的友好接入能力并同时支持 onnx、pb 等主流模型格式的解析与编译。项目地址: https://gitcode.com/cann/ge创作声明：本文部分内容由AI辅助生成（AIGC），仅供参考