[IEEE TMI] Official Implementation for UNet++

Semantic segmentation models with 500+ pretrained convolutional and transformer-based backbones.

Context Axial Reverse Attention Network for Small Medical Objects Segmentation

Polyp-PVT: Polyp Segmentation with Pyramid Vision Transformers, AIR 2023.

PySODEvalToolkit: A Python-based Evaluation Toolbox for Salient Object Detection and Camouflaged Object Detection

PySODMetrics: A Simple and Efficient Implementation of Grayscale/Binary Segmentation Metrcis

[IEEE TIP 2021] COVID-CS Dataset and Code of JCS: An explainable COVID-19 diagnosis system by joint classification and segmentation

Inf-Net: Automatic COVID-19 Lung Infection Segmentation from CT Images, IEEE TMI 2020.

We built a large lung CT scan dataset for COVID-19 by curating data from 7 different public datasets. These datasets have been publicly used in COVID-19 diagnosis literature and proven their effici…

Understanding Convolution for Semantic Segmentation

Enhancing Blind Video Quality Assessment with Rich Quality-aware Features

OpenMMLab Rotated Object Detection Toolbox and Benchmark

Zephyr的小论文，改进YOLOv8模型

YOLOv7+KLD

A PyTorch implementation of MobileNet V2 architecture and pretrained model.

Datasets, Transforms and Models specific to Computer Vision

code and trained models for "Attentional Feature Fusion"

Receptive Field Block Net for Accurate and Fast Object Detection, ECCV 2018

PyTorch implementations of recent Computer Vision tricks (ReXNet, RepVGG, Unet3p, YOLOv4, CIoU loss, AdaBelief, PolyLoss, MobileOne). Other additions: AdEMAMix

label-smooth, amsoftmax, partial-fc, focal-loss, triplet-loss, lovasz-softmax. Maybe useful

A collection of loss functions for medical image segmentation

Collection of common code that's shared among different research projects in FAIR computer vision team.

[ICLR 2024] MogaNet: Efficient Multi-order Gated Aggregation Network

🍀 Pytorch implementation of various Attention Mechanisms, MLP, Re-parameter, Convolution, which is helpful to further understand papers.⭐⭐⭐

Engineering projects related to ELA's attention

This is a warehouse for MobileNetV4-Pytorch-model, can be used to train your image-datasets for vision tasks.

Code for our CVPR2021 paper coordinate attention

Code for ECA-Net: Efficient Channel Attention for Deep Convolutional Neural Networks

The baseline YOLOv8s model has been improved with the integration of Global Attention Mechanism (GAM), Modified Neck, and Wise Intersection over Union (WIoUv3)