提交 8367e974 编写于 作者: 走神的阿圆's avatar 走神的阿圆

modify visual_result

上级 f800b902
......@@ -147,7 +147,7 @@ def predict_mask(module, input_img, id, batch_size, extra=None, r_img=True):
data.update(input_img)
if extra is not None:
data.update(extra)
r_img = False if "r_img" in extra.keys() else True
r_img = False if "visual_result" in extra.keys() else True
results = predict_method(
data=data,
visualization=r_img,
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册