zhanghang1989/ResNeSt

ResNeSt: Split-Attention Networks

What it solves

ResNeSt is a variant of the ResNet architecture designed to improve the performance of computer vision tasks. It specifically targets the limitations of standard ResNets by introducing a split-attention mechanism to better capture cross-channel dependencies and scale-related features.

How it works

The project implements "Split-Attention Networks," which modify the standard residual block to allow the model to learn a more flexible way of aggregating information across different groups of channels. This architecture can be used as a backbone for various downstream tasks such as object detection, instance segmentation, and semantic segmentation.

Who it’s for

This is for computer vision researchers and developers who need a high-performance backbone network for image classification, object detection (e.g., via Mask R-CNN or Cascade R-CNN), and semantic segmentation (e.g., via DeepLabV3).

Highlights

  • Multi-framework support: Provides implementations for PyTorch and MXNet Gluon, with 3rd party versions for TensorFlow, Caffe, and JAX.
  • Downstream boost: Significantly improves performance for models like Mask R-CNN, Cascade R-CNN, and DeepLabV3.
  • Pretrained models: Offers a variety of pretrained weights for different model sizes (ResNeSt-50, 101, 200, and 269).
  • Integration: Includes wrappers for Detectron2 and integration with MMDetection.

Related

  • Project
  • Project
  • Project
  • Project