← all repositories

facebookresearch/maskrcnn-benchmark

A fast, modular PyTorch implementation of object detection and instance segmentation models including Faster R-CNN and Mask R-CNN.

9.4k stars Python Computer VisionML Frameworks
maskrcnn-benchmark
Velocity · 7d
+3.4
★ / day
Trend
steady
star history

This repository provides reference implementations of state-of-the-art instance segmentation and object detection algorithms in PyTorch. It includes optimized RPN, Faster R-CNN, and Mask R-CNN implementations matching or exceeding Detectron accuracy. The library supports multi-GPU training, mixed precision training on NVIDIA tensor cores, batched inference, and CPU inference. It ships with pre-trained models for various Mask R-CNN and Faster R-CNN configurations.

heatdrop uses Google Analytics to see which pages get read — nothing else. Your call. How we handle data.