Mars-craters-challenge Image segmentation on planetary images SSD U-net Both rely on pytorch framework More example in 'Unet example' notebook: