utils.py 文件源码

def get_distanceTransform(mask):
    img = (255*mask).astype(np.uint8)
    dist = cv2.distanceTransform(255-img, cv2.cv.CV_DIST_L2, cv2.cv.CV_DIST_MASK_PRECISE)
    #mask_weights = np.exp(-0.05*dist)
    dist = 1 - dist/np.max(dist)
    return dist