號
物件偵測和實例分割是迄今為止電腦視覺中最重要的應用。然而,在實際使用中,小物體的檢測和大圖像的推理仍然需要改進。 SAHI 來幫助開發人員透過許多視覺實用程式克服這些現實問題。
命令 | 描述 |
---|---|
預測 | 使用任何 ultralytics/mmdet/Detectron2/huggingface/torchvision 模型執行切片/標準視訊/影像預測 |
預測五十一點 | 使用任何 ultralytics/mmdet/ detectorron2/huggingface/torchvision 模型執行切片/標準預測,並在五十一個應用程式中探索結果 |
可可片 | 自動切片COCO註釋和影像文件 |
可可五十一 | 使用按錯誤偵測數量排序的五十個 ui 探索 COCO 資料集上的多個預測結果 |
可可評價 | 針對給定的預測和真實情況評估分類 COCO AP 和 AR |
可可分析 | 計算並導出許多誤差分析圖 |
可可·約洛夫5 | 自動將任何 COCO 資料集轉換為 ultralytics 格式 |
引用 SAHI 的出版品清單(目前 200 多個)
?使用 SAHI 的競賽獲獎者名單
薩希簡介
官方論文(ICIP 2022 口頭)
預訓練權重和 ICIP 2022 紙本文件
使用 FiftyOne 視覺化和評估 SAHI 預測 (2024)(新)
來自“learnopencv.com”的“探索 SAHI”研究文章
「影片教學:用於小物體偵測的切片輔助超推理 - SAHI」(建議)
視頻推理支援上線
卡格爾筆記本
衛星物體偵測
誤差分析圖和評估(建議)
互動式結果視覺化和檢查(建議)
COCO資料集轉換
切片操作筆記本
YOLOX
+ SAHI
演示:(建議)
RT-DETR
+ SAHI
演練:(新)
YOLOv8
+ SAHI
演練:
DeepSparse
+ SAHI
演練:
HuggingFace
+ SAHI
演練:
YOLOv5
+ SAHI
演練:
MMDetection
+ SAHI
演練:
Detectron2
+ SAHI
演練:
TorchVision
+ SAHI
演練:
sahi
: pip install sahi
Shapely
需要透過 Conda 安裝: conda install -c conda-forge shapely
conda install pytorch=1.10.2 torchvision=0.11.3 cudatoolkit=11.3 -c pytorch
conda install pytorch=1.13.1 torchvision=0.14.1 pytorch-cuda=11.7 -c pytorch -c nvidia
pip install yolov5==7.0.13
pip install ultralytics==8.0.207
pip install mim
mim install mmdet==3.0.0
pip install detectron2 -f https://dl.fbaipublicfiles.com/detectron2/wheels/cu113/torch1.10/index.html
pip install transformers timm
pip install super-gradients==3.3.1
在 cli.md 上查找有關sahi predict
命令的詳細資訊。
在視頻推理教程中查找有關視頻推理的詳細資訊。
在 slicing.md 中查找有關圖像/資料集切片實用程式的詳細資訊。
在誤差分析圖和評估中尋找詳細資訊。
在互動式結果視覺化和檢查中尋找詳細資訊。
在 coco.md 上尋找有關 COCO 公用程式(yolov5 轉換、切片、子取樣、過濾、合併、分割)的詳細資訊。
在 mot.md 上尋找有關 MOT 實用程式(地面實況資料集建立、以 mot 挑戰格式匯出追蹤器指標)的詳細資訊。
如果您在工作中使用此包,請將其引用為:
@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}
}
sahi
庫目前支援所有 YOLOv5 模型、MMDetection 模型、Detectron2 模型和 HuggingFace 物件偵測模型。此外,添加新框架也很容易。
您需要做的就是在 sahi/models/ 資料夾下建立一個新的 .py 文件,並在該 .py 檔案中建立一個實作DetectionModel 類別的新類別。您可以將 MMDetection 包裝器或 YOLOv5 包裝器作為參考。
在建立 PR 之前:
pip install -e . " [dev] "
python -m scripts.run_code_style format
法提赫·卡加泰·阿肯
西南·奧努爾·阿爾蒂努克
德夫里姆·恰武什奧盧
傑米爾·森吉茲
奧古爾坎·埃留克塞爾
卡迪爾·納爾
布拉克·馬登
普什帕克·博格
M. 坎 V.
克里斯多福·埃德倫德
伊什沃爾
穆罕默德·埃傑維特
卡迪爾·沙欣
韋伊
永才
阿爾茲貝塔·圖雷科娃
內田壯
權龍惠
內維爾
珍妮·邁拉
克里斯多福·埃德倫德
伊爾克·馬納普
阮氏安
韋吉
艾努蘇蘇茲
普拉納夫·杜萊
拉克謝梅赫拉
卡爾瓊阿萊斯瑪
雅各馬克斯
威廉隆
阿莫·達利瓦爾