open-mmlab / mmdetection

OpenMMLab Detection Toolbox and Benchmark

GitHub Docs

MMDetection

MMDetection is an open source object detection toolbox based on PyTorch. It consists of:

  • Training recipes for object detection and instance segmentation.
  • 360+ pre-trained models to use for fine-tuning (or training afresh).
  • Dataset support for popular vision datasets such as COCO, Cityscapes, LVIS and PASCAL VOC.

Tasks

Choose a task to see what models are available:

Models

Viewing Models for Object Detection:

Models

Viewing Models for Instance Segmentation: