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

Code for Referring Image Segmentation via Cross-Modal Progressive Comprehension, CVPR2020.

Code for "ShineOn: Illuminating Design Choices for Practical Video-based Virtual Clothing Try-on", accepted at WACV 2021 Generation of Human Behavior Workshop.

Implementation for On Provable Benefits of Depth in Training Graph Convolutional Networks

Official source code of paper 'IterMVS: Iterative Probability Estimation for Efficient Multi-View Stereo'

Effective Use of Transformer Networks for Entity Tracking

A framework for Quantification written in Python

A simple rest api serving a deep learning model that classifies human gender based on their faces. (vgg16 transfare learning)

Dist2Dec: A Simplicial Neural Network for Homology Localization

Pytorch implementation of the popular Improv RNN model originally proposed by the Magenta team.

TensorFlow implementation of "Variational Inference with Normalizing Flows"

A script depending on VASP output for calculating Fermi-Softness.

The official implementation of VAENAR-TTS, a VAE based non-autoregressive TTS model.

Proof of concept GnuCash Webinterface

This PyTorch package implements MoEBERT: from BERT to Mixture-of-Experts via Importance-Guided Adaptation (NAACL 2022).

Reference models and tools for Cloud TPUs.

Generic U-Net Tensorflow implementation for image segmentation

Deep Learning and Reinforcement Learning Library for Scientists and Engineers 🔥

Molecular AutoEncoder in PyTorch

SkipGNN: Predicting Molecular Interactions with Skip-Graph Networks (Scientific Reports)

Neural Fixed-Point Acceleration for Convex Optimization