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

Generalized hybrid model for mode-locked laser diodes with an extended passive cavity

Source code for CAST - Crisis Domain Adaptation Using Sequence-to-sequence Transformers (Accepted to ISCRAM 2021, CorePaper).

TransferNet: Learning Transferrable Knowledge for Semantic Segmentation with Deep Convolutional Neural Network

A Pytree Module system for Deep Learning in JAX

A simple code to convert image format and channel as well as resizing and renaming multiple images.

RipsNet: a general architecture for fast and robust estimation of the persistent homology of point clouds

PointPillars inference with TensorRT

IDM: An Intermediate Domain Module for Domain Adaptive Person Re-ID,

A synthetic texture-invariant dataset for object detection of UAVs

Code for Towards Streaming Perception (ECCV 2020) :car:

Serverless proxy for Spark cluster

Supporting code for the Neograd algorithm

Another pytorch implementation of FCN (Fully Convolutional Networks)

Readings for "A Unified View of Relational Deep Learning for Polypharmacy Side Effect, Combination Therapy, and Drug-Drug Interaction Prediction."

This repository contains the code and models necessary to replicate the results of paper: How to Robustify Black-Box ML Models? A Zeroth-Order Optimization Perspective

A Simple Framwork for CV Pre-training Model (SOCO, VirTex, BEiT)

Baleen: Robust Multi-Hop Reasoning at Scale via Condensed Retrieval (NeurIPS'21)

NAACL2021 - COIL Contextualized Lexical Retriever

Benchmark for evaluating open-ended generation

Code for paper "Do Language Models Have Beliefs? Methods for Detecting, Updating, and Visualizing Model Beliefs"