切片辅助超推理-sahi库-get_sliced_prediction使用示例

粼粼淇

已于 2024-10-09 22:16:16 修改

阅读量95

点赞数

文章标签： python numpy pip

于 2024-10-09 22:15:41 首次发布

本文链接：https://blog.csdn.net/lingyunxianhe/article/details/142797766

版权

from sahi import AutoDetectionModel
from sahi.predict import get_prediction, get_sliced_prediction, predict

yolov8_model_path = r'yolov8n.pt'
imfile = r'small-vehicles1.jpeg'
export_dir = r'ashi-res'

detection_model = AutoDetectionModel.from_pretrained(
    model_type='yolov8',
    model_path=yolov8_model_path,
    confidence_threshold=0.3,
    device="cpu",  # or 'cuda:0'
)

def set_cfg_predict():
    slice_height = 320
    slice_width = 320
    result = get_sliced_prediction(
        imfile,
        detection_model,
        slice_height=slice_height,
        slice_width=slice_width,
        overlap_height_ratio=0.2,
        overlap_width_ratio=0.2
    )
    result.export_visuals(export_dir=export_dir, hide_labels=True)

# std_predict()
# auto_cfg_predict()
set_cfg_predict()

上面以yolov8n对工程中示例图为例进行了简单测试。

原图

1、标准推理

def std_predict():
    result = get_prediction(
        imfile,
        detection_model)
    result.export_visuals(export_dir=export_dir, hide_labels=True)

大量漏检

2、自动分割

def auto_cfg_predict():
    result = get_sliced_prediction(
        imfile,
        detection_model
    )
    result.export_visuals(export_dir=export_dir, hide_labels=True)