提交 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): ...@@ -147,7 +147,7 @@ def predict_mask(module, input_img, id, batch_size, extra=None, r_img=True):
data.update(input_img) data.update(input_img)
if extra is not None: if extra is not None:
data.update(extra) 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( results = predict_method(
data=data, data=data,
visualization=r_img, visualization=r_img,
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册