opencv 入门笔记十 图片序列保存为视频

1.代码：

其中：视频格式设置请看这篇：opencv 入门笔记三（采集视频，保存视频）_Newjet666的博客-CSDN博客

其中：读取图片并画框请看这篇：

opencv 入门笔记八 文件夹读取图片序列并视频播放_Newjet666的博客-CSDN博客

#!/usr/bin/python3

import cv2 as cv
import os

# 图片文件夹
dir_path = "/home/newj/下载/Basketball/img/"
# 目标标注框，ground truth
ground_truth_path = "/home/newj/下载/Basketball/groundtruth_rect.txt"
# 视频保存路径
video_save_path = "./imgs2video.MP4"
# 视频编码格式
fourcc = cv.VideoWriter_fourcc(*'mp4v')
# 视频保存相关设置
fps = 15  # 帧率

# 特别注意：图片保存为视频的尺寸要和图片尺寸相等
# 否则程序运行成功，视频无法播放
img_size = (576, 432)  # 尺寸，宽和高

# 实例化视频写入对象
videoWriter = cv.VideoWriter(video_save_path, fourcc, fps, img_size)

# 将文件夹下的所有文件名保存到一个list
imgs_list = os.listdir(dir_path)

# 因为listdir保存到列表，顺序被打乱了
# 按照文件名排序
imgs_list.sort()
# print(np.size(imgs_list))

# 新建文件对象，打开保存有ground truth的txx
f = open(ground_truth_path)
# 按行读取文件，保存为list，list每一个元素为str,之后要进行格式转换
# 其中的一个元素'117,213,34,81\n'
lines = f.readlines()
# print(np.size(lines))

for name, rect in zip(imgs_list, lines):
    # 获取图片路径
    img_path = os.path.join(dir_path, name)
    print(img_path)
    # 读取图片
    img = cv.imread(img_path)
    # 去掉元素末尾的'\n'
    rect = rect.strip('\n')
    # 按','分割字符串为列表，其中每一个元素为字符串
    rect = rect.split(',')
    print(rect)
    # 画框
    img = cv.rectangle(img, (int(rect[0]), int(rect[1])), (int(rect[0]) + int(rect[2]), (int(rect[1]) + int(rect[3]))),
                       (0, 0, 255), 2)
    # 显示
    cv.imshow('Basketball_Sequence', img)
    videoWriter.write(img)
    # 阻塞等待
    # 按下任意键，播放下一帧
    cv.waitKey(0)
videoWriter.release()

2.结果：

basketball play

3.一起进步！