def predict_mask(image_id):
mi = np.load('cache/images/%s_MI.npy' % image_id)
return binary_dilation(mi[1] < 0, disk(3))[np.newaxis, :, :].astype(np.uint8)
微信
公众号
扫码关注公众号