feat: 添加hanoi2物块掉落处理,添加hanoi2 停车延时,修改蓝球计数
This commit is contained in:
27
utils.py
27
utils.py
@@ -190,6 +190,33 @@ class label_filter:
|
||||
return True
|
||||
return False
|
||||
'''
|
||||
description: 查询两个目标 只有 target_label 返回 box
|
||||
param {*} self
|
||||
param {*} tlabel
|
||||
return {[bool]}
|
||||
'''
|
||||
def get_two(self, target_label, label):
|
||||
response = self.get_resp()
|
||||
if response['code'] == 0:
|
||||
ret, results = self.filter_box(response['data'])
|
||||
if ret:
|
||||
expect_boxes = (results[:, 0] == target_label.value)
|
||||
boxes = results[expect_boxes, :]
|
||||
if len(boxes) != 0:
|
||||
target_bool = True
|
||||
target_box = boxes[:, 2:]
|
||||
else:
|
||||
target_bool = False
|
||||
target_box = None
|
||||
expect_boxes = (results[:, 0] == label.value)
|
||||
boxes = results[expect_boxes, :]
|
||||
if len(boxes) != 0:
|
||||
label_bool = True
|
||||
else:
|
||||
label_bool = False
|
||||
return (target_bool, label_bool, target_box)
|
||||
return (False, False, None)
|
||||
'''
|
||||
description: 判断传入的多目标标签是否存在,存在返回 True
|
||||
param {*} self
|
||||
param {*} tlabel
|
||||
|
||||
Reference in New Issue
Block a user