Skydio Interview Question

implement segmentation training and testing code.