yolov5 检测封装

最新推荐文章于 2024-04-02 10:53:05 发布

AI算法网奇

最新推荐文章于 2024-04-02 10:53:05 发布

阅读量753

点赞数

分类专栏： pytorch知识宝典文章标签： YOLO

本文链接：https://blog.csdn.net/jacke121/article/details/135709811

版权

pytorch知识宝典专栏收录该内容

496 篇文章 224 订阅 ¥29.90 ¥99.00

订阅专栏

超级会员免费看

本文介绍了YOLOv5在PyTorch框架下的检测封装过程，并成功进行了v7版本的测试，验证了其在目标检测任务上的有效性和准确性。

摘要由CSDN通过智能技术生成

yolov5 pytorch推理检测封装 v7测试成功


import csv
import os
import platform
import sys
from pathlib import Path

import numpy as np
import torch

FILE = Path(__file__).resolve()
ROOT = FILE.parents[0]  # YOLOv5 root directory
if str(ROOT) not in sys.path:
    sys.path.append(str(ROOT))  # add ROOT to PATH
ROOT = Path(os.path.relpath(ROOT, Path.cwd()))  # relative

from ultralytics.utils.plotting import Annotator, colors, save_one_box

from models.common import DetectMultiBackend
from utils.dataloaders import IMG_FORMATS, VID_FORMATS, LoadImages, LoadScreenshots, LoadStreams
from utils.general import (  cv2,       non_max_suppression,      scale_boxes,    xyxy2xywh)
from utils.torch_utils import select_device, smart_inference_mode

def letterbox_image(img, inp_dim):
    '''resize image with unchanged aspect ratio using padding'''
    img_w, img_h = img.shape[1], img.shape[0]
    w, h = inp_