跟踪结果的mask显示问题

[hilinxinhui]

您好，我想请教一下refinement module的mask输出如何显示在输出的图片上，我在demo.py中的Step 3中调用了RefineModule的get_mask方法得到了mask矩阵，请问如何在输出的图像中显示mask？

我添加的代码是：

""" Step 3: refine tracking results with Alpha-Refine """
mask_bbox = pred_bbox
pred_bbox = RF_module.refine(img, np.array(pred_bbox))
mask = RF_module.get_mask(img, np.array(mask_bbox))

[hilinxinhui]

mask = mask > threshold即可，mask矩阵的元素是[0, 1]内的实数，threshold视情况取值