暂存
fix: 修复task多次开启问题 feat: 新增 891 参数(医院停早,红球靠外) pref: 弃用 action 类
This commit is contained in:
6
utils.py
6
utils.py
@@ -285,7 +285,11 @@ class label_filter:
|
||||
else:
|
||||
target_bool = False
|
||||
target_box = None
|
||||
expect_boxes = (results[:, 0] == label.value)
|
||||
if len(label) == 1:
|
||||
expect_boxes = (results[:, 0] == label[0].value)
|
||||
elif len(label) == 2:
|
||||
expect_boxes = np.logical_or(results[:, 0] == label[0].value, results[:, 0] == label[1].value)
|
||||
|
||||
boxes = results[expect_boxes, :]
|
||||
# 在此处过滤
|
||||
if len(boxes) != 0:
|
||||
|
||||
Reference in New Issue
Block a user