speed.py 文件源码

def segment_speed_image(speed_image_bin):
        speed_image_labeled, num_labels = morphology.label(
            speed_image_bin, background=0, connectivity=1, return_num=True
        )

        segments = []
        for label in range(1, num_labels+1):
            (yy, xx) = np.nonzero(speed_image_labeled == label)
            min_y, max_y = np.min(yy), np.max(yy)
            min_x, max_x = np.min(xx), np.max(xx)
            seg_img = speed_image_bin[min_y:max_y+1, min_x:max_x+1]
            segments.append(SpeedSegment(seg_img))

        return segments