All Projects → greed2411 → unet_pytorch

greed2411 / unet_pytorch

Licence: Apache-2.0 license
Pytorch implementation of UNet for converting aerial satellite images into google maps kinda images.

Programming Languages

python
139335 projects - #7 most used programming language

Projects that are alternatives of or similar to unet pytorch

multiclass-semantic-segmentation
Experiments with UNET/FPN models and cityscapes/kitti datasets [Pytorch]
Stars: ✭ 96 (+255.56%)
Mutual labels:  unet, semantic-segmentation, unet-image-segmentation
satellite-Image-Semantic-Segmentation-Unet-Tensorflow-keras
Collection of different Unet Variant suchas VggUnet, ResUnet, DenseUnet, Unet. AttUnet, MobileNetUnet, NestedUNet, R2AttUNet, R2UNet, SEUnet, scSEUnet, Unet_Xception_ResNetBlock
Stars: ✭ 43 (+59.26%)
Mutual labels:  unet, semantic-segmentation, unet-image-segmentation
ResUNetPlusPlus-with-CRF-and-TTA
ResUNet++, CRF, and TTA for segmentation of medical images (IEEE JBIHI)
Stars: ✭ 98 (+262.96%)
Mutual labels:  unet, semantic-segmentation, unet-image-segmentation
Segmentation models.pytorch
Segmentation models with pretrained backbones. PyTorch.
Stars: ✭ 4,584 (+16877.78%)
Mutual labels:  unet, semantic-segmentation, unet-pytorch
Keras Unet
Helper package with multiple U-Net implementations in Keras as well as useful utility tools helpful when working with image semantic segmentation tasks. This library and underlying tools come from multiple projects I performed working on semantic segmentation tasks
Stars: ✭ 196 (+625.93%)
Mutual labels:  unet, semantic-segmentation
mobilenet segmentation
Binary semantic segmentation with UNet based on MobileNetV2 encoder
Stars: ✭ 18 (-33.33%)
Mutual labels:  semantic-segmentation, unet-image-segmentation
Deep Learning In Production
Develop production ready deep learning code, deploy it and scale it
Stars: ✭ 216 (+700%)
Mutual labels:  unet, semantic-segmentation
Geo Deep Learning
Deep learning applied to georeferenced datasets
Stars: ✭ 91 (+237.04%)
Mutual labels:  unet, semantic-segmentation
image-segmentation
Mask R-CNN, FPN, LinkNet, PSPNet and UNet with multiple backbone architectures support readily available
Stars: ✭ 62 (+129.63%)
Mutual labels:  unet, semantic-segmentation
3D-UNet-PyTorch-Implementation
The implementation of 3D-UNet using PyTorch
Stars: ✭ 78 (+188.89%)
Mutual labels:  unet, semantic-segmentation
celldetection
Cell Detection with PyTorch.
Stars: ✭ 44 (+62.96%)
Mutual labels:  unet, semantic-segmentation
Eye In The Sky
Satellite Image Classification using semantic segmentation methods in deep learning
Stars: ✭ 185 (+585.19%)
Mutual labels:  unet, semantic-segmentation
Semantic Segmentation Of Remote Sensing Images
遥感图像的语义分割,基于深度学习,在Tensorflow框架下,利用TF.Keras,运行环境TF2.0+
Stars: ✭ 125 (+362.96%)
Mutual labels:  unet, semantic-segmentation
Segmentation
Tensorflow implementation : U-net and FCN with global convolution
Stars: ✭ 101 (+274.07%)
Mutual labels:  unet, semantic-segmentation
SmaAt-UNet
PyTorch-Code for the Paper "SmaAt-UNet: Precipitation Nowcasting using a Small, Attentive UNet-Architecture"
Stars: ✭ 112 (+314.81%)
Mutual labels:  unet, unet-pytorch
Polyp-Segmentation-using-UNET-in-TensorFlow-2.0
Implementing polyp segmentation using the U-Net and CVC-612 dataset.
Stars: ✭ 52 (+92.59%)
Mutual labels:  semantic-segmentation, unet-image-segmentation
Unets
Implemenation of UNets for Lung Segmentation
Stars: ✭ 18 (-33.33%)
Mutual labels:  unet, unet-image-segmentation
Multiclass Semantic Segmentation Camvid
Tensorflow 2 implementation of complete pipeline for multiclass image semantic segmentation using UNet, SegNet and FCN32 architectures on Cambridge-driving Labeled Video Database (CamVid) dataset.
Stars: ✭ 67 (+148.15%)
Mutual labels:  unet, semantic-segmentation
Paddleseg
End-to-end image segmentation kit based on PaddlePaddle.
Stars: ✭ 1,244 (+4507.41%)
Mutual labels:  unet, semantic-segmentation
pytorch-UNet
2D and 3D UNet implementation in PyTorch.
Stars: ✭ 107 (+296.3%)
Mutual labels:  unet, semantic-segmentation

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.

Note that the project description data, including the texts, logos, images, and/or trademarks, for each open source project belongs to its rightful owner. If you wish to add or remove any projects, please contact us at [email protected].