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

最新推荐文章于 2024-07-25 16:59:12 发布

m0_67401499

最新推荐文章于 2024-07-25 16:59:12 发布

阅读量2.3k

点赞数 7

分类专栏：面试学习路线阿里巴巴文章标签： python 深度学习计算机视觉机器学习 sklearn

本文链接：https://blog.csdn.net/m0_67401499/article/details/125346054

版权

这篇博客介绍了如何将Yolov5的detect.py转换为一个Python API，以便其他程序可以方便地调用。通过创建detect类，实现了与detect.py相同的功能，并详细说明了类的制作和调用过程，适用于实时处理和图像识别任务。

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

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

作为一个“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