MaskToRotatedRect Mask转旋转矩形

MaskToRotatedRect 用于将 AIDetect 算子推理的 mask_list 和 mask_class_list 转换为旋转矩形。可选择指定的类别输出。

类型	功能
MaskList	将 Mask 列表转换为旋转矩形。
MaskClassList	将 Mask 类别列表转换为旋转矩形。

MaskList

将 MaskToRotatedRect 算子类型选择 MaskList，用于将 Mask 列表转换为旋转矩形。

算子参数

• 选择类别/select_class：当有数据连接到 mask_name_list 端口时，可以输入指定的类别来输出对应的旋转矩形。

• 矩形列表/rect_list：设置旋转矩形列表的曝光属性。

  • 打开曝光。

  • 关闭曝光。

数据信号输入输出

输入：

• mask_list：

  • 数据类型：Image

  • 输入内容：Mask 列表

• mask_names：

  • 数据类型：String

  • 输入内容：Mask 名称列表

说明

该端口连接/不连接，输出所有类别的旋转矩阵。连接时，可填写 select_class 来输出指定类别的旋转矩形。

输出：

• rect_list：

  • 数据类型：RotatedRectList

  • 输出内容：旋转矩形列表

功能演示

使用 MaskToRotatedRect 算子中选择 MaskList，用于将 Mask 列表中 pear 转换为旋转矩形。

步骤1：算子准备

添加 Trigger（2个）、Load 、AIDetectGPU、MaskToRotatedRect 算子至算子图。

步骤2：设置算子参数

1. 设置 Trigger 算子参数：

   • 算子名称 → InitTrigger

   • 类型 → InitTrigger

2. 设置 Load 算子参数：

   • 类型 → Image

   • 文件 → → 选择图像文件名 ( example_data/mask_data_train/20221010141101489/rgb.png )

3. 设置 AIDetectGPU 算子参数：

   • 类型 → MaskRCNN

   • 类名文件路径 → → 选择相应文件名（ example_data/mask_data_train/fruits.txt )

   • 权重文件路径 → → 选择相应权重文件名 （ example_data/mask_data_train/train_output/model_final.pth )

   • 配置文件路径 → → 选择相应配置文件名（ example_data/mask_data_train/train_output/config.yaml )

   • 物体得分阈值 → → 0.75

   • 识别结果图像 → 可视

4. 设置 MaskToRotatedRect 算子参数：

   • 类型 → MaskList

   • 选择列表 → pear

步骤3：连接算子

步骤4：运行

点击 RVS 运行按钮，触发 Trigger 算子。

运行结果

运行结果显示如下，2D 视图中输出 AIDetectGPU 算子的结果。可以看到有 3个梨子，MaskToRotatedRect 算子输出端口输出类别为梨子的旋转矩形列表。

MaskClassList

将 MaskToRotatedRect 算子类型选择 MaskClassList，用于将 Mask 类别列表转换为旋转矩形。

算子参数

• 选择类别/select_class：当有数据连接到 mask_name_list 端口时，可以输入指定的类别来输出对应的旋转矩形。

• 矩阵列表/rect_list：设置旋转矩形列表的曝光属性。

  • 打开曝光。

  • 关闭曝光。

数据信号输入输出

输入：

• mask_class_list：

  • 数据类型：Image

  • 输入内容：Mask 类别列表

• mask_names：

  • 数据类型：String

  • 输入内容：Mask 名称列表

说明

该端口连接/不连接，输出所有类别的旋转矩阵。连接时，可填写 select_class 来输出指定类别的旋转矩形。

输出：

• rect_list：

  • 数据类型：RotatedRectList

  • 输出内容：旋转矩形列表

功能演示

与上述 MaskList 模块类似，请参照该章节功能演示模块。