Stars
Deep Residual Learning for Image Recognition
CVNets: A library for training computer vision networks
Automated, hardware-independent Hand-Eye Calibration
Code release for "Masked-attention Mask Transformer for Universal Image Segmentation"
Official PyTorch implementation of SegFormer
This repo holds the code of TransFuse: Fusing Transformers and CNNs for Medical Image Segmentation
HiFormer: Hierarchical Multi-scale Representations Using Transformers for Medical Image Segmentation (WACV 2023)
Official code for Conformer: Local Features Coupling Global Representations for Visual Recognition
Use of Attention Gates in a Convolutional Neural Network / Medical Image Classification and Segmentation
[ICCV 2023] DETRs with Collaborative Hybrid Assignments Training
UNetFormer: A UNet-like transformer for efficient semantic segmentation of remote sensing urban scene imagery, ISPRS. Also, including other vision transformers and CNNs for satellite, aerial image …
CSWin Transformer: A General Vision Transformer Backbone with Cross-Shaped, CVPR 2022
[ECCV 2024] Official Implementation and Dataset Release for "A Unified Anomaly Synthesis Strategy with Gradient Ascent for Industrial Anomaly Detection and Localization"
Pretrain, finetune ANY AI model of ANY size on multiple GPUs, TPUs with zero code changes.
Detectron2 is a platform for object detection, segmentation and other visual recognition tasks.
PyTorch deep learning projects made easy.
A Guidance on PyTorch Coding Style Based on Kaggle Dogs vs. Cats
Official Implementation of MobileUNETR: A Lightweight End-To-End Hybrid Vision Transformer For Efficient Medical Image Segmentation (ECCV2024) (Oral)
U-Mamba: Enhancing Long-range Dependency for Biomedical Image Segmentation
A novel segmentation model termed Swin UNEt TRansformers (Swin UNETR). Specially for the task of 3D semantic segmentation.
This repository includes the official project of TransUNet, presented in our paper: TransUNet: Transformers Make Strong Encoders for Medical Image Segmentation.
The open source code of SA-UNet: Spatial Attention U-Net for Retinal Vessel Segmentation.