PyTorch implementation of UNet++ (Nested U-Net).

Last update: Jan 04, 2023

Overview

PyTorch implementation of UNet++ (Nested U-Net)

This repository contains code for a image segmentation model based on UNet++: A Nested U-Net Architecture for Medical Image Segmentation implemented in PyTorch.

[NEW] Add support for multi-class segmentation dataset.

[NEW] Add support for PyTorch 1.x.

Requirements

PyTorch 1.x or 0.41

Installation

Create an anaconda environment.

conda create -n=<env_name> python=3.6 anaconda
conda activate <env_name>

Install PyTorch.

conda install pytorch torchvision cudatoolkit=10.1 -c pytorch

Install pip packages.

pip install -r requirements.txt

Training on 2018 Data Science Bowl dataset

Download dataset from here to inputs/ and unzip. The file structure is the following:

inputs
└── data-science-bowl-2018
    ├── stage1_train
    |   ├── 00ae65...
    │   │   ├── images
    │   │   │   └── 00ae65...
    │   │   └── masks
    │   │       └── 00ae65...            
    │   ├── ...
    |
    ...

Preprocess.

python preprocess_dsb2018.py

Train the model.

python train.py --dataset dsb2018_96 --arch NestedUNet

Evaluate.

python val.py --name dsb2018_96_NestedUNet_woDS

(Optional) Using LovaszHingeLoss

Clone LovaszSoftmax from bermanmaxim/LovaszSoftmax.

git clone https://github.com/bermanmaxim/LovaszSoftmax.git

Train the model with LovaszHingeLoss.

python train.py --dataset dsb2018_96 --arch NestedUNet --loss LovaszHingeLoss

Training on original dataset

Make sure to put the files as the following structure (e.g. the number of classes is 2):

inputs
└── <dataset name>
    ├── images
    |   ├── 0a7e06.jpg
    │   ├── 0aab0a.jpg
    │   ├── 0b1761.jpg
    │   ├── ...
    |
    └── masks
        ├── 0
        |   ├── 0a7e06.png
        |   ├── 0aab0a.png
        |   ├── 0b1761.png
        |   ├── ...
        |
        └── 1
            ├── 0a7e06.png
            ├── 0aab0a.png
            ├── 0b1761.png
            ├── ...

Train the model.

python train.py --dataset <dataset name> --arch NestedUNet --img_ext .jpg --mask_ext .png

Evaluate.

python val.py --name <dataset name>_NestedUNet_woDS

Results

DSB2018 (96x96)

Here is the results on DSB2018 dataset (96x96) with LovaszHingeLoss.

Model	IoU	Loss
U-Net	0.839	0.365
Nested U-Net	0.842	0.354
Nested U-Net w/ Deepsupervision	0.843	0.362

PyTorch implementation of UNet++ (Nested U-Net).

Related tags

Overview

PyTorch implementation of UNet++ (Nested U-Net)

Requirements

Installation

Training on 2018 Data Science Bowl dataset

(Optional) Using LovaszHingeLoss

Training on original dataset

Results

DSB2018 (96x96)

Owner

4ui_iurz1

Deep learning models for change detection of remote sensing images

PyTorch implementation for MINE: Continuous-Depth MPI with Neural Radiance Fields

A module for solving and visualizing Schrödinger equation.

Autonomous Ground Vehicle Navigation and Control Simulation Examples in Python

This is the implementation of GGHL (A General Gaussian Heatmap Labeling for Arbitrary-Oriented Object Detection)

50-days-of-Statistics-for-Data-Science - This repository consist of a 50-day program

Code artifacts for the submission "Mind the Gap! A Study on the Transferability of Virtual vs Physical-world Testing of Autonomous Driving Systems"

Learning to Predict Gradients for Semi-Supervised Continual Learning

Unsupervised Video Interpolation using Cycle Consistency

Annotate datasets with a semi-trained or fully trained YOLOv5 model

Reducing Information Bottleneck for Weakly Supervised Semantic Segmentation (NeurIPS 2021)

Generating synthetic mobility data for a realistic population with RNNs to improve utility and privacy

Official repository of my book: "Deep Learning with PyTorch Step-by-Step: A Beginner's Guide"

Utilities to bridge Canvas-generated course rosters with GitLab's API.

Monocular Depth Estimation - Weighted-average prediction from multiple pre-trained depth estimation models

Python script to download the celebA-HQ dataset from google drive

Learning Efficient Online 3D Bin Packing on Packing Configuration Trees

[NeurIPS 2020] Semi-Supervision (Unlabeled Data) & Self-Supervision Improve Class-Imbalanced / Long-Tailed Learning

StarGAN-ZSVC: Unofficial PyTorch Implementation

BasicVSR: The Search for Essential Components in Video Super-Resolution and Beyond