LeViT_UNet

For medical image segmentation

Our model is based on LeViT (https://github.com/facebookresearch/LeViT). You'd better gitclone its codes. Thanks for its great job.
There are three models which can be build directly. They are LeViT_UNet_128s, LeViT_UNet_192, and LeViT_UNet_384. You can build one of the models as it follows:

model = Build_LeViT_UNet_192(num_classes=9, pretrained=True)

model.eval()

output = model(torch.randn(1, 1, 224, 224))

If you use this code for a paper please cite:

@article{LeViT-UNet,
  author    = {Guoping Xu and
               Xingrong Wu and
               Xuan Zhang and
               Xinwei He},
  title     = {LeViT-UNet: Make Faster Encoders with Transformer for Medical Image
               Segmentation},
  journal   = {CoRR},
  volume    = {abs/2107.08623},
  year      = {2021},
  url       = {https://arxiv.org/abs/2107.08623},
  eprinttype = {arXiv},
  eprint    = {2107.08623},
  timestamp = {Thu, 22 Jul 2021 11:14:11 +0200},
  biburl    = {https://dblp.org/rec/journals/corr/abs-2107-08623.bib},
  bibsource = {dblp computer science bibliography, https://dblp.org}
}

For medical image segmentation

Related tags

Overview

LeViT_UNet

Owner

Hyperparameter Optimization for TensorFlow, Keras and PyTorch

GLANet - The code for Global and Local Alignment Networks for Unpaired Image-to-Image Translation arxiv

2021 National Underwater Robotics Vision Optics

Deep Implicit Moving Least-Squares Functions for 3D Reconstruction

Learn about quantum computing and algorithm on quantum computing

Camera ready code repo for the NeuRIPS 2021 paper: "Impression learning: Online representation learning with synaptic plasticity".

SVG Icon processing tool for C++

A decent AI that solves daily Wordle puzzles. Works with different websites with similar wordlists,.

Implements an infinite sum of poisson-weighted convolutions

A large dataset of 100k Google Satellite and matching Map images, resembling pix2pix's Google Maps dataset.

A curated list of programmatic weak supervision papers and resources

This is the official repository of XVFI (eXtreme Video Frame Interpolation)

Forecasting Nonverbal Social Signals during Dyadic Interactions with Generative Adversarial Neural Networks

code for Fast Point Cloud Registration with Optimal Transport

FTIR-Deep Learning - FTIR Deep Learning With Python

Punctuation Restoration using Transformer Models for High-and Low-Resource Languages

A python library to artfully visualize Factorio Blueprints and an interactive web demo for using it.

This is the official implementation of VaxNeRF (Voxel-Accelearated NeRF).

Experiments on continual learning from a stream of pretrained models.

Exploring whether attention is necessary for vision transformers