Atlas 200I DK目标检测与追踪技术记录

数据集

数据集采用MOT系列，MOT是多目标追踪常用数据集，MOT数据集对数据集进行了分帧，如要获得视频，需要先利用opencv里的cv2.VideoWriter模块便利图片文件夹，具体代码如下：

import os
import cv2img_path = './img1' #图片存放路径# 读取第一张图片的宽度、高度和通道数（用于设置视频尺寸）
first_image = cv2.imread(os.path.join(img_path, os.listdir(img_path)[0]))
height, width, channels = first_image.shape# 视频保存路径和文件名
output_path = './output.mp4'# 视频帧率
fps = 30# 设置视频编解码器
fourcc = cv2.VideoWriter_fourcc(*'mp4v')# 创建视频写入器
video_writer = cv2.VideoWriter(output_path, fourcc, fps, (width, height))# 遍历图片文件夹中的每张图片，并逐帧写入视频
for filename in sorted(os.listdir(img_path)):img_path = './img1' img_path = os.path.join(img_path, filename)frame = cv2.imread(img_path)# 将当前帧写入视频video_writer.write(frame)# 释放视频写入器
video_writer.release()