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

The goal of the exercises below is to evaluate the candidate knowledge and problem solving expertise regarding the main development focuses for the iFood ML Platform team: MLOps and Feature Store development.

Rethinking Nearest Neighbors for Visual Classification

Official PyTorch implementation and pretrained models of the paper Self-Supervised Classification Network

My implementation of Image Inpainting - A deep learning Inpainting model

Official implementation for NIPS'17 paper: PredRNN: Recurrent Neural Networks for Predictive Learning Using Spatiotemporal LSTMs.

UMPNet: Universal Manipulation Policy Network for Articulated Objects

Generic template to bootstrap your PyTorch project with PyTorch Lightning, Hydra, W&B, and DVC.

A Collection of LiDAR-Camera-Calibration Papers, Toolboxes and Notes

Wav2Vec for speech recognition, classification, and audio classification

a delightful machine learning tool that allows you to train, test and use models without writing code

An experiment on the performance of homemade Q-learning AIs in Agar.io depending on their state representation and available actions

SOLO and SOLOv2 for instance segmentation, ECCV 2020 & NeurIPS 2020.

Fastquant - Backtest and optimize your trading strategies with only 3 lines of code!

This framework implements the data poisoning method found in the paper Adversarial Examples Make Strong Poisons

🚀 PyTorch Implementation of "Progressive Distillation for Fast Sampling of Diffusion Models(v-diffusion)"

Parris, the automated infrastructure setup tool for machine learning algorithms.

Official Codes for Graph Modularity:Towards Understanding the Cross-Layer Transition of Feature Representations in Deep Neural Networks.

Code for the paper: Hierarchical Reinforcement Learning With Timed Subgoals, published at NeurIPS 2021

OpenMMLab's Next Generation Video Understanding Toolbox and Benchmark

LAnguage Model Analysis