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

Official code for "On the Frequency Bias of Generative Models", NeurIPS 2021

a pytorch implementation of auto-punctuation learned character by character

Deep Crop Rotation

Automated Evidence Collection for Fake News Detection

Code for the Weighted, Accelerated and Restarted Primal-dual algorithm. This algorithm achieves stable linear convergence for reconstruction from undersampled noisy measurements under an approximate sharpness condition. See the paper for details.

Implementation of "Fast and Flexible Temporal Point Processes with Triangular Maps" (Oral @ NeurIPS 2020)

SSD: Single Shot MultiBox Detector pytorch implementation focusing on simplicity

Session-aware Item-combination Recommendation with Transformer Network

A C implementation for creating 2D voronoi diagrams

Code for the paper: "On the Bottleneck of Graph Neural Networks and Its Practical Implications"

Classification of ecg datas for disease detection

This tool uses Deep Learning to help you draw and write with your hand and webcam.

Multiview 3D object detection on MultiviewC dataset through moft3d.

Try out deep learning models online on Google Colab

Pytorch implementation for "Distribution-Balanced Loss for Multi-Label Classification in Long-Tailed Datasets" (ECCV 2020 Spotlight)

GLANet - The code for Global and Local Alignment Networks for Unpaired Image-to-Image Translation arxiv

Official Implementation of Few-shot Visual Relationship Co-localization

Official Pytorch implementation of "Beyond Static Features for Temporally Consistent 3D Human Pose and Shape from a Video", CVPR 2021

GANSketchingJittor - Implementation of Sketch Your Own GAN in Jittor

Spiking Neural Network for Computer Vision using SpikingJelly framework and Pytorch-Lightning