MaskToRotatedRect Mask转旋转矩形

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

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

MaskList

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

算子参数

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

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

    • icon_visOn 打开曝光。

    • icon_visOff 关闭曝光。

数据信号输入输出

输入:

  • 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

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

  3. 设置 AIDetectGPU 算子参数:

    • 类型 → MaskRCNN

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

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

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

    • 物体得分阈值 →icon_more → 0.75

    • 识别结果图像 → icon_visOn 可视

  4. 设置 MaskToRotatedRect 算子参数:

    • 类型 → MaskList

    • 选择列表 → pear

步骤3:连接算子

image-20230601173213548

步骤4:运行

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

运行结果

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

image-20230601173544556

image-20230601173446229

MaskClassList

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

算子参数

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

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

    • icon_visOn 打开曝光。

    • icon_visOff 关闭曝光。

数据信号输入输出

输入:

  • mask_class_list

    • 数据类型:Image

    • 输入内容:Mask 类别列表

  • mask_names

    • 数据类型:String

    • 输入内容:Mask 名称列表

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

输出:

  • rect_list

    • 数据类型:RotatedRectList

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

功能演示

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