跳转至
SAHI logo

SAHI: 切片辅助高效推理

用于大规模目标检测和实例分割的轻量级视觉库

teaser

下载量 月下载量 许可证 PyPI 版本 Conda 版本 持续集成
Context7 MCP llms.txt DOI arXiv 在 Colab 打开 HuggingFace Spaces

什么是 SAHI?

SAHI(Slicing Aided Hyper Inference,切片辅助高效推理)是一个提供了通用的切片辅助推理与微调流程,专门用于小目标检测的开源框架。在监控等应用场景中,检测小目标或远处物体是一项重大挑战,因为它们仅由少量像素表示,缺乏足够的细节,传统检测器难以识别。

SAHI 通过一种独特的方法解决了这一问题,该方法可以与任意目标检测器结合使用,而无需额外的微调。在 Visdrone 和 xView 航拍目标检测数据集上的实验评估表明,SAHI 可以使 FCOS、VFNet 和 TOOD 检测器的 AP 分别提高 6.8%、5.1% 和 5.3%。在结合切片辅助微调后,精度可进一步提升,累计提升分别达到 12.7%、13.4% 和 14.5%。该技术已成功集成到 Ultralytics YOLOv8Ultralytics YOLO11Ultralytics YOLO26、HuggingFace Transformers、RT-DETR、TorchVision、MMDetection、Detectron2、YOLOv5、YOLOE、YOLO-World 和 Roboflow RF-DETR 等模型中。

  • 快速开始


    使用 pip 安装 sahi,几分钟即可上手。

    快速开始

  • 工作原理


    了解切片算法、使用场景和调优方法。

    切片推理

  • 模型集成


    将 SAHI 与 Ultralytics、HuggingFace、MMDetection、TorchVision 等框架配合使用。

    所有模型

  • 预测


    使用 SAHI 对新图片、视频和流进行预测。

    了解更多

  • 切片


    学习如何对大图像和数据集进行切片以进行推理。

    了解更多

  • COCO 工具


    处理 COCO 格式数据集,包括创建、拆分和过滤。

    了解更多

  • 命令行工具


    通过命令行使用 SAHI 进行预测和数据集操作。

    了解更多

  • FiftyOne


    交互式探索和比较检测结果。

    了解更多

  • Notebooks


    为所有支持的框架提供动手实践的 Colab notebooks。

    浏览 notebooks