python+opencv目标跟踪之鼠标

怼烎@ 2023-10-01 09:23 11阅读 0赞

## 项目介绍 ##

#### 目标：跟踪鼠标的移动，一开始采用深度学习的方法，但是效果不好，所以采用了下面的方法 ####

#### 操作描述：鼠标框住一个目标，然后按回车，框会跟随目标 ####

#### 本代码可以在任意视频跟踪所框定的目标 ####

import cv2
    import sys
    
    (major_ver, minor_ver, subminor_ver) = (cv2.__version__).split('.')
    print(major_ver, minor_ver, subminor_ver)
    
    if __name__ == '__main__':
        # 创建跟踪器
        tracker_type = 'Mouse '
        tracker = cv2.TrackerMIL_create()
        # 读入视频
        video = cv2.VideoCapture('./mouse2.avi')
        fourcc = cv2.VideoWriter_fourcc(*'XVID')
        fps = video.get(cv2.CAP_PROP_FPS)
        size = (int(video.get(cv2.CAP_PROP_FRAME_WIDTH)), int(video.get(cv2.CAP_PROP_FRAME_HEIGHT)))
        out = cv2.VideoWriter('mouse_tracking.avi', fourcc, 10.0, size)
        # 读入第一帧
        ok, frame = video.read()
        if not ok:
            print('Cannot read video file')
            sys.exit()
        # 定义一个bounding box
        bbox = (287, 23, 86, 320)
        bbox = cv2.selectROI(frame, False)
        # 用第一帧初始化
        ok = tracker.init(frame, bbox)
    
        while True:
            ok, frame = video.read()
            if not ok:
                break
            # Start timer
            timer = cv2.getTickCount()
            # Update tracker
            ok, bbox = tracker.update(frame)
            # Cakculate FPS
            fps = cv2.getTickFrequency() / (cv2.getTickCount() - timer)
            # Draw bonding box
            if ok:
                p1 = (int(bbox[0]), int(bbox[1]))
                p2 = (int(bbox[0] + bbox[2]), int(bbox[1] + bbox[3]))
                cv2.rectangle(frame, p1, p2, (255, 0, 0), 2, 1)
            else:
                cv2.putText(frame, "Tracking failed detected", (100, 80), cv2.FONT_HERSHEY_SIMPLEX, 0.75, (0, 0, 255), 2)
            # 展示tracker类型
            cv2.putText(frame, tracker_type + "Tracker", (100, 20), cv2.FONT_HERSHEY_SIMPLEX, 0.75, (50, 170, 50), 2)
            # 展示FPS
            cv2.putText(frame, "FPS:" + str(fps), (100, 50), cv2.FONT_HERSHEY_SIMPLEX, 0.75, (50, 170, 50), 2)
            # Result
            out.write(frame)
            cv2.imshow("Tracking", frame)
            # Exit
            k = cv2.waitKey(1) & 0xff
            if k == 27: break

### 效果如下： ###

mouse