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

[EMNLP 2021] MuVER: Improving First-Stage Entity Retrieval with Multi-View Entity Representations

Pytorch implementation of PTNet for high-resolution and longitudinal infant MRI synthesis

Project page for our ICCV 2021 paper "The Way to my Heart is through Contrastive Learning"

COVID-VIT: Classification of Covid-19 from CT chest images based on vision transformer models

Temporally Coherent GAN SIGGRAPH project.

This library is a location of the LegacyLogger for PyTorch Lightning.

PAIRED in PyTorch 🔥

Wordle-solver - Wordle answer generation program in python

Bounding Wasserstein distance with couplings

Code for EMNLP 2021 paper: "Learning Implicit Sentiment in Aspect-based Sentiment Analysis with Supervised Contrastive Pre-Training"

[arXiv'22] Panoptic NeRF: 3D-to-2D Label Transfer for Panoptic Urban Scene Segmentation

Code for Mesh Convolution Using a Learned Kernel Basis

MoCoPnet - Deformable 3D Convolution for Video Super-Resolution

Code for "Learning the Best Pooling Strategy for Visual Semantic Embedding", CVPR 2021

[ICCV'21] NEAT: Neural Attention Fields for End-to-End Autonomous Driving

Human pose estimation from video plays a critical role in various applications such as quantifying physical exercises, sign language recognition, and full-body gesture control.

Using fully convolutional networks for semantic segmentation with caffe for the cityscapes dataset

CAUSE: Causality from AttribUtions on Sequence of Events

a basic code repository for basic task in CV(classification,detection,segmentation)

Neuron class provides LNU (Linear Neural Unit), QNU (Quadratic Neural Unit), RBF (Radial Basis Function), MLP (Multi Layer Perceptron), MLP-ELM (Multi Layer Perceptron - Extreme Learning Machine) neurons learned with Gradient descent or LeLevenberg–Marquardt algorithm