prepare-images.py 文件源码

def compute_filters(data):
    filter_data = np.zeros((1, data.shape[1], data.shape[2]), dtype=np.float32)
    filter_data[0] = sobel(np.clip(data[0] / 600.0, 0, 1)) + sobel(np.clip(data[1] / 600.0, 0, 1)) + sobel(np.clip(data[2] / 600.0, 0, 1))

    return filter_data