- Notifications
You must be signed in to change notification settings - Fork666
Framework agnostic sliced/tiled inference + interactive ui + error analysis plots
License
obss/sahi
Folders and files
Name | Name | Last commit message | Last commit date | |
---|---|---|---|---|
Repository files navigation
SAHI helps developers overcome real-world challenges in object detection by enablingsliced inference for detecting small objects in large images. It supports various popular detection models and provides easy-to-use APIs.
Command | Description |
---|---|
predict | perform sliced/standard video/image prediction using anyultralytics/mmdet/huggingface/torchvision model - seeCLI guide |
predict-fiftyone | perform sliced/standard prediction using any supported model and explore results infiftyone app -learn more |
coco slice | automatically slice COCO annotation and image files - seeslicing utilities |
coco fiftyone | explore multiple prediction results on your COCO dataset withfiftyone ui ordered by number of misdetections |
coco evaluate | evaluate classwise COCO AP and AR for given predictions and ground truth - checkCOCO utilities |
coco analyse | calculate and export many error analysis plots - see thecomplete guide |
coco yolo | automatically convert any COCO dataset toultralytics format |
📜 List of publications that cite SAHI (currently 400+)
🏆 List of competition winners that used SAHI
SAHI's documentation isindexed in Context7 MCP, providing AI coding assistants with up-to-date, version-specific code examples and API references. We also provide anllms.txt file following the emerging standard for AI-readable documentation. To integrate SAHI docs with your AI development workflow, check out theContext7 MCP installation guide.
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.11.21
- Install your desired detection framework (mmdet):
pip install mimmim install mmdet==3.3.0
- Install your desired detection framework (roboflow):
pip install inference>=0.50.3 rfdetr>=1.1.0
Introduction to SAHI - explore thecomplete documentation for advanced usage
Official paper (ICIP 2022 oral)
2025 Video Tutorial (RECOMMENDED)
'VIDEO TUTORIAL: Slicing Aided Hyper Inference for Small Object Detection - SAHI'
Error analysis plots & evaluation (RECOMMENDED)
Interactive result visualization and inspection (RECOMMENDED)
Find detailed info on usingsahi predict
command in theCLI documentation and explore theprediction API for advanced usage.
Find detailed info on video inference atvideo inference tutorial.
Find detailed info atError Analysis Plots & Evaluation.
ExploreFiftyOne integration for interactive visualization and inspection.
Check thecomprehensive COCO utilities guide for YOLO conversion, dataset slicing, subsampling, filtering, merging, and splitting operations. Learn more about theslicing utilities for detailed control over image and dataset slicing parameters.
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}}
We welcome contributions! Please see ourContributing Guide to get started. Thank you 🙏 to all our contributors!
About
Framework agnostic sliced/tiled inference + interactive ui + error analysis plots
Topics
Resources
License
Uh oh!
There was an error while loading.Please reload this page.
Stars
Watchers
Forks
Uh oh!
There was an error while loading.Please reload this page.