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

Original code for "Zero-Shot Domain Adaptation with a Physics Prior"

Implemenets the Contourlet-CNN as described in C-CNN: Contourlet Convolutional Neural Networks, using PyTorch

Accelerated Multi-Modal MR Imaging with Transformers

This repository contains the map content ontology used in narrative cartography

A Low Complexity Speech Enhancement Framework for Full-Band Audio (48kHz) based on Deep Filtering.

PyImpetus is a Markov Blanket based feature subset selection algorithm that considers features both separately and together as a group in order to provide not just the best set of features but also the best combination of features

Official PyTorch Implementation for InfoSwap: Information Bottleneck Disentanglement for Identity Swapping

A flexible tool for creating, organizing, and sharing visualizations of live, rich data. Supports Torch and Numpy.

Face and other object detection using OpenCV and ML Yolo

Python PID Tuner - Based on a FOPDT model obtained using a Open Loop Process Reaction Curve

GPT-Code-Clippy (GPT-CC) is an open source version of GitHub Copilot

Homepage of paper: Paint Transformer: Feed Forward Neural Painting with Stroke Prediction, ICCV 2021.

Semi-SDP Semi-supervised parser for semantic dependency parsing.

Learning hidden low dimensional dyanmics using a Generalized Onsager Principle and neural networks

Simple machine learning library / 簡單易用的機器學習套件

Symmetry and Uncertainty-Aware Object SLAM for 6DoF Object Pose Estimation

COCO Style Dataset Generator GUI

An auto discord account and token generator. Automatically verifies the phone number. Works without proxy. Bypasses captcha.

An official implementation of the Anchor DETR.

Minimal implementation and experiments of "No-Transaction Band Network: A Neural Network Architecture for Efficient Deep Hedging".