Repository files navigation

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

I used the pix2pix maps dataset available overhere

• pytorch:0.4.1
• torchvision:0.2.1

GeneratedLabelInput

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 usingcheckpoint and I found adiscussion post related to it. Haven't tried it yet, therefore any suggestion or crtiques are always welcome.