fix: 修復hanoi2右側圖像上下翻轉時過濾條件錯誤的情況
pref: 6_9較好參數 pref: 關閉hanoi2循跡模型切換
This commit is contained in:
62
utils.py
62
utils.py
@@ -264,33 +264,41 @@ class label_filter:
|
||||
return (target_bool, label_bool, target_box)
|
||||
return (False, False, None)
|
||||
|
||||
# '''
|
||||
# description: 查询两个目标 只有 target_label 返回 box
|
||||
# param {*} self
|
||||
# param {*} tlabel
|
||||
# return {[bool]}
|
||||
# '''
|
||||
# def get_two_hanoi(self, target_label, label, ymax_range):
|
||||
# response = self.get_resp()
|
||||
# if response['code'] == 0:
|
||||
# ret, results = self.filter_box_custom(response['data'], ymax_range)
|
||||
# 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: 查询两个目标 只有 target_label 返回 box
|
||||
param {*} self
|
||||
param {*} tlabel
|
||||
return {[bool]}
|
||||
'''
|
||||
def get_two_hanoi(self, target_label, label, flipv):
|
||||
response = self.get_resp()
|
||||
if response['code'] == 0:
|
||||
# FIXME 直接在外部过滤,不在 fliter 内过滤
|
||||
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:
|
||||
# 如果垂直翻转 (走右侧) 且 *ymin* 小于 60(走右侧)
|
||||
if flipv:
|
||||
label_bool = all(box[3] > 60 for box in boxes)
|
||||
# 如果不垂直翻转 (走左侧) 且 *ymax* 大于 180(走左侧)
|
||||
else:
|
||||
label_bool = all(box[5] < 180 for box in boxes)
|
||||
# label_bool = True
|
||||
else:
|
||||
label_bool = False
|
||||
return (target_bool, label_bool, target_box)
|
||||
return (False, False, None)
|
||||
|
||||
'''
|
||||
description: 判断传入的多目标标签是否存在,存在返回 True
|
||||
|
||||
Reference in New Issue
Block a user