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

Leon嘞2023-10-18 12:50

数据集

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

python 复制代码

import os
import cv2
 
img_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()