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

A PyTorch implementation of ViTGAN based on paper ViTGAN: Training GANs with Vision Transformers.

Attentive Implicit Representation Networks (AIR-Nets)

Deep Learning & 3D Convolutional Neural Networks for Speaker Verification

A SAT-based sudoku solver

CondenseNet: Light weighted CNN for mobile devices

Commonality in Natural Images Rescues GANs: Pretraining GANs with Generic and Privacy-free Synthetic Data - Official PyTorch Implementation (CVPR 2022)

A convolutional recurrent neural network for classifying A/B phases in EEG signals recorded for sleep analysis.

ISBI 2022: Cross-level Contrastive Learning and Consistency Constraint for Semi-supervised Medical Image.

Code base for the paper "Scalable One-Pass Optimisation of High-Dimensional Weight-Update Hyperparameters by Implicit Differentiation"

Automatic voice-synthetised summaries of latest research papers on arXiv

Code to reproduce the results for Statistically Robust Neural Network Classification, published in UAI 2021

BanditPAM: Almost Linear-Time k-Medoids Clustering

coldcuts is an R package to automatically generate and plot segmentation drawings in R

Pull sensitive data from users on windows including discord tokens and chrome data.

Second Order Optimization and Curvature Estimation with K-FAC in JAX.

Spectral Tensor Train Parameterization of Deep Learning Layers

Jittor Medical Segmentation Lib -- The assignment of Pattern Recognition course (2021 Spring) in Tsinghua University

Pytorch implementation of CVPR2020 paper “VectorNet: Encoding HD Maps and Agent Dynamics from Vectorized Representation”

Pytorch code for ICRA'21 paper: "Hierarchical Cross-Modal Agent for Robotics Vision-and-Language Navigation"

code for paper"A High-precision Semantic Segmentation Method Combining Adversarial Learning and Attention Mechanism"