zoukankan      html  css  js  c++  java
  • 机器学习进阶-目标跟踪-KCF目标跟踪方法 1.cv2.multiTracker_create(构造选框集合) 2. cv2.TrackerKCF_create(获得KCF追踪器) 3. cv2.resize(变化图像大小) 4.cv2.selectROI(在图像上框出选框)

    1. tracker = cv2.multiTracker_create() 获得追踪的初始化结果

    2.cv2.TrackerKCF_create() 获得KCF追踪器

    3.cv2.resize(frame, (w, h), cv2.INTER_AEAR)  # 进行图像大小的重新变化
    参数说明:frame表示输入图片,(w, h) 表示变化后的长和宽, cv2.INTER_AEAR表示插值的方法

    4.cv2.selectROI(‘Frame’, frame, fromCenter=False,showCrosshair) 在图像上进行选框操作
    参数说明:‘Frame'表示图片的名字,frame表示读取的图片,fromCenter表示选框的起点是否在中心位置,showCrosshair表示是否展示选框

    KCF是一种鉴别式追踪方法,这类方法一般在追踪过程中训练一个目标检测器,使用目标检测器与预测下一个阶段预测位置是否是目标,然后使用新检测结果去跟新训练集而跟新目标检测其

    KCF贡献

    1.使用脊回归训练目标检测器,成功利用循环矩阵在傅里叶空间对角化的性质,将矩阵的运算转换为向量的Hadamad积,即元素的点乘,大大降低了运算量,提高了运算速度

        1.脊回归,在原有的基础上引入了λ|w^2| 为了防止过拟合的出现,即w的分布更小且更加均匀(0, 0, 0, 1) 和 (0.25, 0.25, 0.25, 0.25), 对w进行求导,导数为0,解得w

       2.循环矩阵,即训练图像都是通过目标样本的循环位移得到的, 通过PxQ进行x轴和y轴的平移变化

       

        3.傅氏空间矩阵对角化, 即对上述的循环空间做一个傅里叶变化

         4. 将循环矩阵x带入到脊回归进行求解

    最后做的一个总结部分

    参考链接:https://www.cnblogs.com/YiXiaoZhou/p/5925019.html

    代码:

    第一步: 进行 argparse参数设置

    第二步:使用字典构造已有的追踪目标的方法

    第三步:使用cv2.multiTracker_create() 构造出追踪器集合trackers

    第四步:使用cv2.capture进行视频的读入

    第五步:while True循环,使用cv2.read() 读入图片

    第六步:使用cv2.resize进行图片的放缩变换

    第七步:trackers.apply(frame)对图片运用追踪算法

    第八步:循环boxes对图像进行画图操作
    第九步:展示图片,

    第十步:并判断按键,如果按s,进行box的添加,如果按ESC键就退出

    第十一步:如果按s,使用box = cv2.selectROI()进行区域的选择,并使用trackers.add将追踪器和图片,以及区域进行添加

    import argparse
    import cv2
    import time
    
    # 第一步进行参数设置
    ap = argparse.ArgumentParser()
    ap.add_argument('-v', '--video', type=str, default='videos/soccer_01.mp4',
                    help='path to input video file')
    ap.add_argument('-t', '--tracker', type=str,
                    default='kcf', help='Opencv object tracker type')
    
    args = vars(ap.parse_args())
    
    
    # opencv已经实现的追踪算法
    # 第二步:构造cv2已有追踪算法的列表
    OPENCV_OBJECT_TRACKERS = {
        "csrt": cv2.TrackerCSRT_create,
        "kcf": cv2.TrackerKCF_create,
        "boosting": cv2.TrackerBoosting_create,
        "mil": cv2.TrackerMIL_create,
        "tld": cv2.TrackerTLD_create,
        "medianflow": cv2.TrackerMedianFlow_create,
        "mosse": cv2.TrackerMOSSE_create
    }
    
    # 第三步:实例化追踪器
    # 实例化Opencv's multi-object tracker
    trackers = cv2.MultiTracker_create()
    
    # 第四步:使用cv2.VideoCapture读取视频
    vs = cv2.VideoCapture(args['video'])
    
    while True:
        # 第五步:读入第一张图片
        frame = vs.read()
        frame = frame[1]
        # 到头了就结束
        if frame is None:
            break
    
        # 第六步:使用cv2.resize对图像进行长宽的放缩操作
        h, w = frame.shape[:2]
        width = 600
        r = width / float(w)
        dim = (width, int(r * h))
        frame = cv2.resize(frame, dim, cv2.INTER_AREA)
    
        # 第七步:使用trackers.apply获得矩形框
        (success, boxes) = trackers.update(frame)
    
        # 第八步:循环多组矩形框,进行画图操作
        for box in boxes:
            (x, y, w, h) = [int(v) for v in box]
            cv2.rectangle(frame, (x, y), (x+w, y+h), (0, 255, 0), 2)
    
        # 第九步:进行图像展示
        cv2.imshow('Frame', frame)
    
        # 第十步:判断按键,如果是s的话,进行画出新的box
        key = cv2.waitKey(100) & 0xff
    
        if key == ord('s'):
            # 第十一步:选择一个区域,按s键,并将tracker追踪器,frame和box传入到trackers中
            box = cv2.selectROI('Frame', frame, fromCenter=False,
                                showCrosshair=True)
            tracker = OPENCV_OBJECT_TRACKERS[args['tracker']]()
            trackers.add(tracker, frame, box)
    
        elif key == 27:
            break
    
    
    vs.release()
    cv2.destroyAllWindows()

      

                                        区域选择                                                                                                   追踪效果

  • 相关阅读:
    SpringBoot整合MyBatis【数据库连接】
    SpringBoot整合日志
    SpringBoot引用lombok让代码更简洁
    AOP集成log4j日志
    SpringBoot全局捕获异常
    SpringBoot整合Freemarker
    springboot配置请求跨域问题
    Maven工程的pom文件引用本地jar包
    使用Mybatis插件generator自动生成代码
    SET NOCOUNT 的用法
  • 原文地址:https://www.cnblogs.com/my-love-is-python/p/10452242.html
Copyright © 2011-2022 走看看