如果在识别任务配置文件中有类型为‘deform object’的识别任务,如(taskid为4的识别任务类型为‘deform object’),那么我们在Agent端会收到此任务的识别结果,deform object类型的结果示例如下面所示。其中‘‘groupID'’,”frameSeq“,”image“ ,”deviceIndex“。是Agent收到的GameReg进程发过来的图像帧结果的通用字段,”result“对应的是识别结果字段,存储识别结果的数据结构为字典,字典的key值为整型taskid,此值和识别任务的配置文件是一致的(如上文提到的任务的taskid为4)。字典的value值为对应的此任务的识别结果。一个识别任务(task),可能会有多个元素(Element)组成,所以每个识别任务的结果是个列表,结果列表中的元素对应Element的识别结果。
{
'groupID': int,
'frameSeq': int,
'image': mat,
'deviceIndex': int,
'result':
{
// taskID: result
1:
[
{
'flag': bool,
'boxes':
[
{
'tmplName': str,
'score': float,
'scale': float,
'classID': int,
'x': int,
'y': int,
'w': int,
'h': int
}
]
}
]
}
}
GameReg进程把识别结果发送到共享内存中,Agent进程通过平台内置的AgentAPIMgr接收结果包,解析结果包中的类型为‘deform object'的任务的主要代码示例如下。
DEFORM_OBJECT_TASK_ID = 4
class GameXXXEnv(GameEnv):
def __init__(self):
GameEnv.__init__(self)
self.__agentAPI = None
def Init(self):
self.__agentAPI = AgentAPIMgr.AgentAPIMgr()
ret = self.__agentAPI.Initialize(taskCfgFile, referCfgFile)
if not ret:
self.logger.error('Agent API Init Failed')
return False
ret = self.__agentAPI.SendCmd(AgentAPIMgr.MSG_SEND_GROUP_ID, 1)
if not ret:
self.logger.error('send message failed')
return False
# 其他内容省略
return True
# 获取类型为'deform object'的任务的结果
def GetDeformObjectState(self, taskResult):
state = False
boxes = list()
if taskResult is None:
return state, boxes
for item in taskResult:
# 如果当前Element没有识别到目标物,则flag为False。否则为True.
state = item['flag']
if state:
for n in range(len(item['boxes'])):
# 获取目标物的位置('x', 'y', 'w', 'h',单位:像素)和类别(整型)
x = item['boxes'][n]['x']
y = item['boxes'][n]['y']
w = item['boxes'][n]['w']
h = item['boxes'][n]['h']
classID = item['boxes'][n]['classID']
px = int(x + w / 2)
py = int(y + h / 2)
boxes.append([classID, px, py])
return state, boxes
def GetGameInfo(self):
while True:
# 通过api接口,获取当前帧的处理结果
gameInfo = self.__agentAPI.GetInfo(AgentAPIMgr.GAME_RESULT_INFO)
if gameInfo is None:
continue
# 获取所有识别任务的结果
result = gameInfo['result']
if result is None:
continue
# 获取DEFORM_OBJECT_TASK_ID对应的任务的结果
deformObjResult = result.get(DEFORM_OBJECT_TASK_ID)
# 解析DEFORM_OBJECT_TASK_ID对应的任务的结果
state, boxes = self.GetDeformObjectState(deformObjResult)
# 根据需求做相应的处理
break
def IsEpisodeStart(self):
return True
def IsEpisodeOver(self):
return False