【pysot 深入理解】 一、demo.py

加入pysot包。

# --------------------- 查找包 ------------------------
import sys
sys.path.append('D:\code\Github_code\pysot')
# --------------------- 查找包 ------------------------

函数说明

脚本使用了 OpenCV 的 selectROI 函数进行初始对象选择，这需要在图形界面环境中运行。

get_frames 函数：

从视频文件或摄像头中读取帧。
如果 video_name 是空，它从摄像头读取帧。
如果 video_name 是视频文件（以 .avi 或 .mp4 结尾），它从该文件读取帧。
如果 video_name 是一个目录，它会读取目录中的所有 .jpg 或 .jpeg 图像文件。

main 函数：

加载配置和模型。
创建跟踪器实例。
通过 get_frames 函数循环获取每一帧。
对第一帧使用 selectROI 来手动选择跟踪对象的初始位置。
对后续帧使用跟踪器来跟踪对象。
显示跟踪结果，并在每帧上绘制跟踪对象的边界框或多边形。

修改

本地相机

args.video_name 参数为空或者设置为 0。在大多数情况下，0 是默认的摄像头索引，它通常指向计算机的内置摄像头或第一个连接的外部摄像头。