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

fix ci

上级 969d43f6
......@@ -149,7 +149,10 @@ def predict_mask(module, input_img, id, batch_size, extra=None, r_img=True):
data.update(extra)
r_img = False if "r_img" in extra.keys() else True
results = predict_method(
data=data, visualization=r_img, use_gpu=use_gpu, batch_size=batch_size)
data=data,
visualization=r_img,
use_gpu=use_gpu,
batch_size=batch_size)
results = utils.handle_mask_results(results)
except Exception as err:
curr = time.strftime("%Y-%m-%d %H:%M:%S", time.localtime(time.time()))
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册