yolov5 python API（供其他程序调用）

你的yolov5??是否只局限于detect.py？如果其他程序要调用yolov5，就需要制作一个detect.py的python API。python无处不对象，制作detect API实际上就是制作detect类。

目录

• 前言
• 一、总体思路
• 二、制作detect类
• 二、调用detect类
• 结语

---

前言

yolov5源码版本：截止2022.2.3
链接：https://github.com/ultralytics/yolov5

作为一个“CV”主义者，在此之前在各平台都没有找到合适的API代码。其中有一篇不错的文章https://www.pythonheidong.com/blog/article/851830/44a42d351037d307d02d/
可惜代码版本过于“久远”，部分函数已经不适用了。本文以一种简单粗暴的方式制作与detect.py功能一样的API，即使源码更新，按照我的方法也能快速制作一个API供其他程序调用。

一、总体思路

其他程序调用yolo，实际上就是把图像传给detect.py。为了最大化实现detect.py的所有功能，最直接的方式是摄像头或者视频流把帧图像存储在‘date/images’目录中，然后把帧图像从‘runs/detect/exp’中读取出来。这种方法增加了处理时间，不过实测存储和读取图像这部分的延迟很低，即便是在树莓派上。

二、制作detect类

在detect.py中添加以下代码

class DetectAPI:
    def __init__(self, weights='weights/yolov5s.pt', data='data/coco128.yaml', imgsz=None, conf_thres=0.25,
                 iou_thres=0.45, max_det=1000, device='0', view_img=False, save_txt=False,
                 save_conf=False, save_crop=False, nosave=False, classes=None, agnostic_nms=False, augment=False,
                 visualize=False, update=False, project='runs/detect', name='myexp', exist_ok=False, line_thickness=3,
                 hide_labels=False, hide_conf=False, half=False, dnn=False):

        if imgsz is None:
            self.imgsz = [640, 640]
        self.weights = weights
        self.data = data
        self.source = 'data/myimages'
        self.imgsz = [640, 640]
        self.conf_thres = conf_thres
        self.iou_thres = iou_thres
        self.max_det = max_det
        self.device = device
        self.view_img = view_img
        self.save_txt = save_txt
        self.save_con