V社区-机器智能技术交流-自定义算法模块获取输入直线的方法

自定义算法模块获取输入直线的方法

1442
0
分享

二维码

分享链接
2022-12-19 10:49

VM算法平台

自定义算法模块如何获取输入直线，从而扩展到直线边缘缺陷检测等算法领域？

1. 在算法模块生成器的自定义输入输出中配置输入直线参数

编译界面工程、算法工程，并将文件拷贝到工具箱目录后，模块界面如下图所示。

输入直线的参数名称从模块.xml中查看，子元素类型均为float。

2. 在算法工程中获取输入直线的数据

首先在AlgorithmModule.h文件中定义输入直线的起点、终点变量，示例代码如下：

float lineStartPX;
float lineStartPY;
float lineEndPX;
float lineEndPY;

然后在AlgorithmModule.cpp文件的Process函数中获取输入直线的数据，示例代码如下：

       int     nRet = IMVS_EC_UNKNOWN;
	HKA_CHECK_ERROR((IMVS_NULL == hInput || IMVS_NULL == hOutput), IMVS_EC_PARAM);

	HKA_F32 fValue = 0;
	int nArrayCount = 0;

	nRet = VM_M_GetFloat(hInput, "lineStartPX", 0, &fValue, &nArrayCount);
	if (IMVS_EC_OK == nRet && nArrayCount > 0)
	{
		lineStartPX = fValue;
	}
	HKA_MODU_CHECK_ERROR(IMVS_EC_OK != nRet, nRet);
	
	VM_M_GetFloat(hInput, "lineStartPY", 0, &fValue, &nArrayCount);
	if (IMVS_EC_OK == nRet && nArrayCount > 0)
	{
		lineStartPY = fValue;
	}
	HKA_MODU_CHECK_ERROR(IMVS_EC_OK != nRet, nRet);	

	VM_M_GetFloat(hInput, "lineEndPX", 0, &fValue, &nArrayCount);
	if (IMVS_EC_OK == nRet && nArrayCount > 0)
	{
		lineEndPX = fValue;
	}
	HKA_MODU_CHECK_ERROR(IMVS_EC_OK != nRet, nRet);	

	VM_M_GetFloat(hInput, "lineEndPY", 0, &fValue, &nArrayCount);
	if (IMVS_EC_OK == nRet && nArrayCount > 0)
	{
		lineEndPY = fValue;
	}
	HKA_MODU_CHECK_ERROR(IMVS_EC_OK != nRet, nRet);

至此，实现模块输入直线的配置与数据获取。