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

PyTorch reimplementation of the paper Involution: Inverting the Inherence of Convolution for Visual Recognition [CVPR 2021].

Package to compute Mauve, a similarity score between neural text and human text. Install with `pip install mauve-text`.

A PyTorch implementation of SIN: Superpixel Interpolation Network

The official code repository for examples in the O'Reilly book 'Generative Deep Learning'

Find-Lane-Line - Use openCV library and Python to detect the road-lane-line

ThunderGBM: Fast GBDTs and Random Forests on GPUs

PSML: A Multi-scale Time-series Dataset for Machine Learning in Decarbonized Energy Grids

A scikit-learn compatible neural network library that wraps PyTorch

This repository contains the entire code for our work "Two-Timescale End-to-End Learning for Channel Acquisition and Hybrid Precoding"

Line-level Handwritten Text Recognition (HTR) system implemented with TensorFlow.

A Comparative Framework for Multimodal Recommender Systems

3D-CariGAN: An End-to-End Solution to 3D Caricature Generation from Normal Face Photos

Code for paper "Learning to Reweight Examples for Robust Deep Learning"

Zero-shot Learning by Generating Task-specific Adapters

Hard cater examples from Hopper ICLR paper

PyTorch implementation of Deformable Convolution

Some experiments with tennis player aging curves using Hilbert space GPs in PyMC. Only experimental for now.

A transformer model to predict pathogenic mutations

PyTorch implementation of federated learning framework based on the acceleration of global momentum

This repository contains the code for our paper VDA (public in EMNLP2021 main conference)