pysot系列--demo.py笔记（迭代器获取视频帧）

最新推荐文章于 2024-07-02 17:20:59 发布

原创

最新推荐文章于 2024-07-02 17:20:59 发布 · 654 阅读

4 ·

CC 4.0 BY-SA版权

本文详细记录了使用pysot库通过迭代器方式获取视频帧的操作过程，适合已具备一定计算机视觉基础的读者进行深入学习。

需要具备的先验知识

# 获取图像文件夹的所有图片文件名字
glob.glob(path+"*p")

# 获取感兴趣区域，并返回区域的矩形框坐标
# selectROI(windowName, img, showCrosshair=None, fromCenter=None):
    #. 参数windowName：选择的区域被显示在的窗口的名字
    #. 参数img：要在什么图片上选择ROI
    #. 参数showCrosshair：是否在矩形框里画十字线.
    #. 参数fromCenter：是否是从矩形框的中心开始画
image=cv2.imread('example.png')
cv2.namedWindow('img')
r = cv2.selectROI('roi', image, False, False )
print(r)
cv2.waitKey(0)
#########
# 返回的是一个列表[min_x,min_y,w,h]：也就是左上角的坐标与区域宽高

from __future__ import absolute_import
from __future__ import division
from __future__ import print_function
from __future__ import unicode_literals
import sys
#sys.path.append('你的\\pysot-master\\路径')
sys.path.append('C:\\Users\\lpf\\Desktop\\Code_of_Paper\\pysot-master\\pysot')
sys.path.append('C:\\Users\\lpf\\Desktop\\Code_of_Paper\\pysot-master')
import os
import argparse

import cv2
import torch
import numpy as np
from glob import glob

from pysot.core.config import cfg
from pysot.models.model_builder import ModelBuilder
from pysot.tracker.tracker_builder import build_tracker

# python tools/demo.py \
config = r'../experiments/siamrpn_r50_l234_dwxcorr/config.yaml'
snapshot = 'C:/Users/lpf/Desktop/Code_of_Paper/PySOT/siamrpn_r50_l234_dwxcorr/model.pth'
# 摄像头输入视频
# 注释掉video即可

# 没有摄像头，使用本地视频
# video = r"../demo/bag.avi" # (in case you don't have webcam)  
# video = r"C:\Users\lpf\Desktop\Code_of_Paper\1_MyLearning\8_KITTI\project10.avi"

# 没有本地视频，使用本地图片，jpg或者png都可以，