probabilistic_unet

Segmentation model

Reimplementation of a neural network model for conditional segmentation of ambiguous images

A U-Net combined with a variational auto-encoder that is able to learn conditional distributions over semantic segmentations.

GitHub

546 stars
20 watching
100 forks
Language: Jupyter Notebook
last commit: over 1 year ago
Linked from 1 awesome list

generative-modelsneuripsneurips-2018nipsnips-2018semantic-segmentationu-netvariational-autoencodersvariational-inference

Backlinks from these awesome lists:

Related projects:

Repository Description Stars
4uiiurz1/pytorch-nested-unet An implementation of a deep learning model for image segmentation using PyTorch 851
zfturbo/zf_unet_224_pretrained_model A pre-trained convolutional neural network model for image segmentation tasks. 214
preritj/segmentation Deep learning models for semantic segmentation of images 100
ethanhe42/u-net A convolutional neural network architecture for biomedical image segmentation 426
ternaus/ternausnet An image segmentation model using an UNet architecture with a pre-trained VGG11 encoder. 1,047
andreaazzini/segnet.tf An implementation of a segmentation network in TensorFlow for image classification and object detection tasks. 66
fabianbormann/tensorflow-deconvnet-segmentation An implementation of a deep learning algorithm for image segmentation using convolutional neural networks 220
scottykwok/cervix-roi-segmentation-by-unet This project provides an implementation of the U-NET algorithm for segmenting regions of interest in cervical images. 35
tkuanlun350/tensorflow-segnet A TensorFlow-based implementation of the SegNet segmentation algorithm, with modifications to address index unravel issues and support multiple features such as dilation and multi-resolution learning. 369
nv-tlabs/gscnn This code implements a neural network architecture designed to perform semantic segmentation in computer vision tasks. 920
jakeret/tf_unet An open source U-Net image segmentation implementation using Tensorflow 1,899
ecobost/cnn4brca This project uses Convolutional Neural Networks (CNN) for image segmentation of breast cancer lesions. 75
wasidennis/adaptsegnet This project implements a deep learning-based approach to adapt semantic segmentation models from one domain to another. 849
ternaus/iglovikov_segmentation A deep learning-based semantic segmentation pipeline using the Catalyst framework. 20
openai/pixel-cnn A generative model with tractable likelihood and easy sampling, allowing for efficient data generation. 1,921