python opencv自动跟踪移动目标（DaSiamRPN）

最新推荐文章于 2024-10-29 12:38:56 发布

原创

最新推荐文章于 2024-10-29 12:38:56 发布 · 2.7k 阅读

11 ·

CC 4.0 BY-SA版权

本文介绍了使用Python结合OpenCV库，通过DaSiamRPN网络进行目标跟踪的方法。程序运行稳定，跟踪效果良好，且具有较高的帧率。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

1、代码有几个bug，cv2.drawContours函数返回值要注意

2、程序调用了DaSiamRPN跟踪网络，跟踪效果不错，fps也很高

3、代码如下

'''
读取视频、检测前景目标，调用DiameseRPN进行跟踪
'''
import cv2
import torch
import numpy as np
from os.path import realpath, dirname, join
from net import SiamRPNvot
from run_SiamRPN import SiamRPN_init, SiamRPN_track
from utils import get_axis_aligned_bbox, cxy_wh_2_rect



# load net
net = SiamRPNvot()
net.load_state_dict(torch.load(join(realpath(dirname(__file__)), 'SiamRPNVOT.model')))
net.eval().cuda()


def videoTrack():
    video_path = " "
    cap = cv2.VideoCapture(0)
    ret,frame = cap.read()
    fgbg = cv2.createBackgroundSubtractorMOG2()

    startTrack = False
    restartTrack = False
    stopTrack = False
    isTracking = False

    while(ret):
        #前景检测
        fgmask = fgbg.apply(frame)
        element = cv2.getStructuringElement(cv2.MORPH_RECT,(3,3))

        #前景处理
        fgmask = cv2.erode(fgmask,element)
        masked = cv