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

House-GAN++: Generative Adversarial Layout Refinement Network towards Intelligent Computational Agent for Professional Architects

Our solution for SSN Invente 2021's Hackathon

Finite-temperature variational Monte Carlo calculation of uniform electron gas using neural canonical transformation.

Dungeons and Dragons randomized content generator

DirectVoxGO reconstructs a scene representation from a set of calibrated images capturing the scene.

(CVPR 2022) Energy-based Latent Aligner for Incremental Learning

📚 A collection of Jupyter notebooks for learning and experimenting with OpenVINO 👓

An implementation of MobileFormer

Newt - a Gaussian process library in JAX.

Optimizing DR with hard negatives and achieving SOTA first-stage retrieval performance on TREC DL Track (SIGIR 2021 Full Paper).

Implementation and replication of ProGen, Language Modeling for Protein Generation, in Jax

Elucidating Robust Learning with Uncertainty-Aware Corruption Pattern Estimation

Code for T-Few from "Few-Shot Parameter-Efficient Fine-Tuning is Better and Cheaper than In-Context Learning"

Official implementation of the ICML2021 paper "Elastic Graph Neural Networks"

SFD implement with pytorch

Efficient-GlobalPointer - Pytorch Efficient GlobalPointer

Server files for UltimateLabeling

Stochastic Downsampling for Cost-Adjustable Inference and Improved Regularization in Convolutional Networks

This is an official implementation for "ResT: An Efficient Transformer for Visual Recognition".

Learning trajectory representations using self-supervision and programmatic supervision.