Voltar ao ranking

obss/sahi

Pythonobss.github.io/sahi/

Framework agnostic sliced/tiled inference + interactive ui + error analysis plots

object-detectioninstance-segmentationcomputer-visionsmall-object-detectionlarge-imagepytorchpythoncocodeep-learningmachine-learningremote-sensinghuggingface
Crescimento de estrelas
Estrelas
5.4k
Forks
766
Crescimento semanal
Issues
0
2k4k
jan. de 2021nov. de 2022set. de 2024jul. de 2026
ArtefatosPyPIpip install sahi
README
SAHI logo

SAHI: Slicing Aided Hyper Inference

A lightweight vision library for performing large scale object detection & instance segmentation

teaser

Total Downloads Monthly Downloads PyPI Version Conda Version License
CI Known Vulnerabilities CodeFactor DOI
Context7 MCP llms.txt DeepWiki HuggingFace Spaces

Overview

SAHI helps developers overcome real-world challenges in object detection by enabling sliced inference for detecting small objects in large images. It supports various popular detection models and provides easy-to-use APIs.

🌐 English | 🇨🇳 简体中文

Command Description
predict Perform sliced/standard video/image prediction using any ultralytics / mmdet / huggingface / torchvision model — see CLI guide
predict-fiftyone Perform sliced/standard prediction using any supported model and explore results in fiftyone applearn more
coco slice Automatically slice COCO annotation and image files — see slicing utilities
coco fiftyone Explore multiple prediction results on your COCO dataset with fiftyone ui ordered by number of misdetections
coco evaluate Evaluate classwise COCO AP and AR for given predictions and ground truth — check COCO utilities
coco analyse Calculate and export many error analysis plots — see the complete guide
coco yolo Automatically convert any COCO dataset to ultralytics format

Approved by the Community

📜 List of publications that cite SAHI (currently 600+)

🏆 List of competition winners that used SAHI

Approved by AI Tools

SAHI's documentation is indexed in Context7 MCP, providing AI coding assistants with up-to-date, version-specific code examples and API references. We also provide an llms.txt file following the emerging standard for AI-readable documentation. To integrate SAHI docs with your AI development workflow, check out the Context7 MCP installation guide.

Installation

Basic Installation

pip install sahi
Detailed Installation (Click to open)
  • Install your desired version of pytorch and torchvision:
pip install torch==2.7.0 torchvision==0.22.0 --index-url https://download.pytorch.org/whl/cu126

(torch 2.1.2 is required for mmdet support):

pip install torch==2.1.2 torchvision==0.16.2 --index-url https://download.pytorch.org/whl/cu121
  • Install your desired detection framework (ultralytics):
pip install ultralytics>=8.3.161
  • Install your desired detection framework (huggingface):
pip install transformers>=4.49.0 timm
  • Install your desired detection framework (yolov5):
pip install yolov5==7.0.14 sahi==0.12.1
  • Install your desired detection framework (mmdet):
pip install mim
mim install mmdet==3.3.0
  • Install your desired detection framework (roboflow):
pip install inference>=0.51.5 rfdetr>=1.6.2

Quick Start

Learning Resources

Resource Type
Introduction to SAHI Blog Post
2025 Video Tutorial Video
Official Paper (ICIP 2022 oral) Paper
Pretrained Weights & ICIP 2022 Paper Files Benchmark
Visualizing and Evaluating SAHI Predictions with FiftyOne Blog Post
Exploring SAHI – learnopencv.com Article
Slicing Aided Hyper Inference Explained by Encord Article
Video Tutorial: SAHI for Small Object Detection Video
Satellite Object Detection Blog Post
COCO Dataset Conversion Blog Post
Kaggle Notebook Notebook
Error Analysis Plots & Evaluation Discussion
Interactive Result Visualization and Inspection Discussion
Video Inference Support Discussion
Slicing Operation Notebook Notebook
Complete Documentation Docs

Notebooks & Demos

Framework Notebook Demo
YOLO26 Open In Colab
YOLO11 Open In Colab
YOLO11-OBB Open In Colab
YOLOE Open In Colab
Roboflow / RF-DETR Open In Colab
RT-DETR v2 Open In Colab
RT-DETR Open In Colab
HuggingFace Open In Colab
GroundingDINO Open In Colab
YOLOv5 Open In Colab
MMDetection Open In Colab
Detectron2 Open In Colab
TorchVision Open In Colab
YOLOX HuggingFace Spaces

sahi-yolox

Framework Agnostic Sliced/Standard Prediction

sahi-predict

Find detailed info on using sahi predict command in the CLI documentation and explore the prediction API for advanced usage.

Find detailed info on video inference at video inference tutorial.

Error Analysis Plots & Evaluation

sahi-analyse

Find detailed info at Error Analysis Plots & Evaluation.

Interactive Visualization & Inspection

sahi-fiftyone

Explore FiftyOne integration for interactive visualization and inspection.

Other Utilities

Check the comprehensive COCO utilities guide for YOLO conversion, dataset slicing, subsampling, filtering, merging, and splitting operations. Learn more about the slicing utilities for detailed control over image and dataset slicing parameters.

Citation

If you use this package in your work, please cite as:

@article{akyon2022sahi,
  title={Slicing Aided Hyper Inference and Fine-tuning for Small Object Detection},
  author={Akyon, Fatih Cagatay and Altinuc, Sinan Onur and Temizel, Alptekin},
  journal={2022 IEEE International Conference on Image Processing (ICIP)},
  doi={10.1109/ICIP46576.2022.9897990},
  pages={966-970},
  year={2022}
}
@software{obss2021sahi,
  author       = {Akyon, Fatih Cagatay and Cengiz, Cemil and Altinuc, Sinan Onur and Cavusoglu, Devrim and Sahin, Kadir and Eryuksel, Ogulcan},
  title        = {{SAHI: A lightweight vision library for performing large scale object detection and instance segmentation}},
  month        = nov,
  year         = 2021,
  publisher    = {Zenodo},
  doi          = {10.5281/zenodo.5718950},
  url          = {https://doi.org/10.5281/zenodo.5718950}
}

Contributing

We welcome contributions! Please see our Contributing Guide to get started. Thank you 🙏 to all our contributors!

Repositórios relacionados
ultralytics/ultralytics

Ultralytics YOLO26, YOLO11, YOLOv8 — object detection, instance segmentation, semantic segmentation, image classification, pose estimation, object tracking

PythonPyPIGNU Affero General Public License v3.0ultralyticsyolov8
platform.ultralytics.com
59.7k11.4k
ultralytics/yolov5

Ultralytics YOLOv5 in PyTorch for object detection, instance segmentation, classification, training, and export.

PythonPyPIGNU Affero General Public License v3.0yolov5object-detection
docs.ultralytics.com/yolov5/
57.7k17.5k
roboflow/supervision

We write your reusable computer vision tools. 💜

PythonPyPIMIT Licensecomputer-visionimage-processing
supervision.roboflow.com
48.3k4.4k
mudler/LocalAI

LocalAI is the open-source AI engine. Run any model - LLMs, vision, voice, image, video - on any hardware. No GPU required.

GoGo ModulesMIT Licensellamaai
localai.io
47.7k4.3k
blakeblackshear/frigate

NVR with realtime local object detection for IP cameras

TypeScriptnpmMIT Licensertsprealtime
frigate.video
34.5k3.4k
open-mmlab/mmdetection

OpenMMLab Detection Toolbox and Benchmark

PythonPyPIApache License 2.0object-detectioninstance-segmentation
mmdetection.readthedocs.io
32.8k9.8k
WZMIAOMIAO/deep-learning-for-image-processing

deep learning for image processing including classification and object-detection etc.

PythonPyPIGNU General Public License v3.0pytorchtensorflow2
26.3k8.2k
matterport/Mask_RCNN

Mask R-CNN for object detection and instance segmentation on Keras and TensorFlow

PythonPyPIOthermask-rcnntensorflow
25.6k11.6k
amusi/CVPR2026-Papers-with-Code

CVPR 2026 论文和开源项目合集

cvprcvpr2020
22.8k2.8k
amusi/CVPR2025-Papers-with-Code

CVPR 2025 论文和开源项目合集

cvprcvpr2020
19.1k2.6k
amusi/CVPR2024-Papers-with-Code

CVPR 2024 论文和开源项目合集

cvprcvpr2020
18.7k2.6k
cvat-ai/cvat

Computer Vision Annotation Tool (CVAT) is a leading platform for building high-quality visual datasets for vision AI. It offers open-source, cloud, and enterprise products, as well as labeling services, for image, video, and 3D annotation with AI-assisted labeling, quality assurance, team collaboration, analytics, and developer APIs.

PythonPyPIMIT Licensevideo-annotationcomputer-vision
cvat.ai
16.4k3.8k