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

Learning where to learn - Gradient sparsity in meta and continual learning

Contrastive Language-Image Pretraining

Towards uncontrained hand-object reconstruction from RGB videos

CRNN With PyTorch

All course materials for the Zero to Mastery Deep Learning with TensorFlow course.

Code for "Multi-Time Attention Networks for Irregularly Sampled Time Series", ICLR 2021.

Official Pytorch Implementation of 'Learning Action Completeness from Points for Weakly-supervised Temporal Action Localization' (ICCV-21 Oral)

An unofficial PyTorch implementation of a federated learning algorithm, FedAvg.

[ACL-IJCNLP 2021] "EarlyBERT: Efficient BERT Training via Early-bird Lottery Tickets"

[NeurIPS 2021] Introspective Distillation for Robust Question Answering

SSD: A Unified Framework for Self-Supervised Outlier Detection [ICLR 2021]

A new GCN model for Point Cloud Analyse

(Py)TOD: Tensor-based Outlier Detection, A General GPU-Accelerated Framework

VQGAN+CLIP Colab Notebook with user-friendly interface.

DRLib：A concise deep reinforcement learning library, integrating HER and PER for almost off policy RL algos.

SAT Project - The first project I had done at General Assembly, performed EDA, data cleaning and created data visualizations

scikit-learn inspired API for CRFsuite

an implementation of softmax splatting for differentiable forward warping using PyTorch

kapre: Keras Audio Preprocessors

The Unsupervised Reinforcement Learning Benchmark (URLB)