ImageUtils.py 文件源码

def prepare_expected_mask(mask_path):
    im = cv2.resize(cv2.imread(mask_path, cv2.IMREAD_GRAYSCALE), (output_mask_size, output_mask_size)).astype(np.float32)
    # replace visible color with 1 (actual mask)
    im[im > 0] = 1
    # 0 -> -1
    im[im == 0] = -1
    return im