lxy51

pytorch-image-models

The largest collection of PyTorch image encoders / backbones. Including train, eval, inference, export scripts, and pretrained weights -- ResNet, ResNeXT, EfficientNet, NFNet, Vision Transformer (ViT), MobileNetV4, MobileNet-V3 & V2, RegNet, DPN, CSPNet, Swin Transformer, MaxViT, CoAtNet, ConvNeXt, and more

BCDU-Net

BCDU-Net : Medical Image Segmentation

UNet-Version

ResUnet

Pytorch implementation of ResUnet and ResUnet ++

MultiResUNet

MultiResUNet : Rethinking the U-Net architecture for multimodal biomedical image segmentation

UCTransNet

Implementation of our AAAI'22 work: 'UCTransNet: Rethinking the Skip Connections in U-Net from a Channel-wise Perspective with Transformer'.

LightM-UNet

Pytorch implementation of "LightM-UNet: Mamba Assists in Lightweight UNet for Medical Image Segmentation"

iFormer

iFormer: Inception Transformer

Swin-UMamba

Swin-UMamba: Mamba-based UNet with ImageNet-based pretraining

ACC-UNet

ACC-UNet is A Completely Convolutional UNet model inspired from transformer-based UNets

event_flow

Self-Supervised Learning of Event-based Optical Flow with Spiking Neural Networks

3D-UCaps

3D-UCaps: 3D Capsules Unet for Volumetric Image Segmentation (MICCAI 2021)

MedicalSeg

3D Medical Image Segmentation Models，集成各种医学图像分割模型的小框架，主要是3D，持续更新...

SCUNet-plusplus

[WACV 2024] SCUNet++: Swin-UNet and CNN Bottleneck Hybrid Architecture with Multi-Fusion Dense Skip Connection for Pulmonary Embolism CT Image Segmentation

SA-ConvLSTM-Pytorch

Pytorch implementation of Self-Attention ConvLSTM

LMa-UNet

Large Window-based Mamba UNet for Medical Image Segmentation: Beyond Convolution and Self-attention

AttentionAugmentedConvLSTM

Implementation of TAAConvLSTM and SAAConvLSTM used in "Attention Augmented ConvLSTM for Environment Prediction"

ScaleLong

The official repository of paper "ScaleLong: Towards More Stable Training of Diffusion Model via Scaling Network Long Skip Connection" (NeurIPS 2023)

QG-ConvLSTM

The project page of the paper: Ren Yang, Xiaoyan Sun, Mai Xu and Wenjun Zeng, "Quality-Gated Convolutional LSTM for Enhancing Compressed Video", in IEEE International Conference on Multimedia and Expo (ICME), 2019.

PMFSNet

DAR-UNet

[JBHI2022] A novel 3D unsupervised domain adaptation framework for cross-modality medical image segmentation

LeViT-UNet

For medical image segmentation

BuildingExtraction

SDSC-UNet and DSAT-Net (IEEE GRSL 2023)

BiTr-Unet

This repo is the source code for [BiTr-Unet: a CNN-Transformer Combined Network for MRI Brain Tumor Segmentation].

MultiResUNet

MultiResUNet implementation in PyTorch; MultiResUNet: Rethinking the U-Net Architecture for Multimodal

Pulmonary-Vessel-Segmentation-and-Artery-vein-Separation

The code of paper "Transformer-based 3D U-Net for Pulmonary Vessel Segmentation and Artery-vein Separation from CT images"

IrradianceNet

AMFU-net

Code for "Lightweight Infrared Small Target Detection Network Using Full-Scale Skip Connection U-Net" in IEEE GRSL 2023

MultiResUnet3D

MultiResUnet3D implemented by Pytorch

TACLNet

A Topological-Attention ConvLSTM Network and Its Application to EM Images, MICCAI 2021