/unet_pytorch

Pytorch implementation of UNet for converting aerial satellite images into google maps kinda images.

Primary LanguagePythonApache License 2.0Apache-2.0

Semantic Segmentation using U-Net

Pytorch implementation of U-Net: Convolutional Networks for Biomedical Image Segmentation for segmentation of aerial maps into google maps.

Dataset

I used the pix2pix maps dataset available over here

Requirements

Examples

Generated gen_image_99_400 gen_image_99_0 Label label_image_99_400 label_image_99_0 Input original_image_99_400 original_image_99_0

P.S: It took 12 hours to train on a 1050Ti with a batch size of 5 for 100 epochs. If I tried to increase the batch size, I ran out of memory. I asked around and people suggested using checkpoint and I found a discussion post related to it. Haven't tried it yet, therefore any suggestion or crtiques are always welcome.