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

Official Implementation of SimIPU: Simple 2D Image and 3D Point Cloud Unsupervised Pre-Training for Spatial-Aware Visual Representations

A Lightweight Face Recognition and Facial Attribute Analysis (Age, Gender, Emotion and Race) Library for Python

DexterRedTool - Dexter's Red Team Tool that creates cronjob/task scheduler to consistently creates users

68 keypoint annotations for COFW test data

YOLO-v5 기반 단안 카메라의 영상을 활용해 차간 거리를 일정하게 유지하며 주행하는 Adaptive Cruise Control 기능 구현

Code for the Interspeech 2021 paper "AST: Audio Spectrogram Transformer".

Image-popularity-score - A novel deep regression method for image scoring.

An experimentation and research platform to investigate the interaction of automated agents in an abstract simulated network environments.

Disentangled Lifespan Face Synthesis

Microsoft Cognitive Toolkit (CNTK), an open source deep-learning toolkit

Optimizaciones incrementales al problema N-Body con el fin de evaluar y comparar las prestaciones de los traductores de Python en el ámbito de HPC.

Translate darknet to tensorflow. Load trained weights, retrain/fine-tune using tensorflow, export constant graph def to mobile devices

List of all dependencies affected by node-ipc malicious commit

Hippocampal segmentation using the UNet network for each axis

An educational resource to help anyone learn deep reinforcement learning.

This is a Python wrapper for TA-LIB based on Cython instead of SWIG.

Chunkmogrify: Real image inversion via Segments

Certified Patch Robustness via Smoothed Vision Transformers

Implementation of several Bayesian multi-target tracking algorithms, including Poisson multi-Bernoulli mixture filters for sets of targets and sets of trajectories. The repository also includes the GOSPA metric and a metric for sets of trajectories to evaluate performance.

[CVPR 2021] Rethinking Semantic Segmentation from a Sequence-to-Sequence Perspective with Transformers