SAHI_Yolov8:小目标检测

最新推荐文章于 2024-08-09 17:07:51 发布

犟小孩

最新推荐文章于 2024-08-09 17:07:51 发布

阅读量664

点赞数 2

分类专栏：技术文档文章标签： YOLO 目标检测人工智能

本文链接：https://blog.csdn.net/qq_55542491/article/details/133926375

版权

懒得推gitee，记录一下

import os
import time
import shutil
os.getcwd()

# arrange an instance segmentation model for test
from sahi.utils.yolov8 import (
    download_yolov8s_model,
)
import cv2
from sahi import AutoDetectionModel
from sahi.utils.cv import read_image
from sahi.utils.file import download_from_url
from sahi.predict import get_prediction, get_sliced_prediction, predict
from IPython.display import Image


detection_model = AutoDetectionModel.from_pretrained(
    model_type='yolov8',
    # YOLOv8模型的路径
    model_path="/home/hxzh/PycharmProjects/SAHI_YOLOv8/best.pt",
    # YOLOv8模型的路径
    confidence_threshold=0.3,
    # 设备类型。
    # 如果您的计算机配备 NVIDIA GPU，则可以通过将 'device' 标志更改为'cuda:0'来启用 CUDA 加速；否则，将其保留为'cpu'
    device="cuda:0", # or 'cuda:0'
)
start_time=time.time()
images_path="/media/hxzh/D/Dataset/plean_test/0926"
for i in os.listdir(im