代码如下:
import os
import cv2
def extract_last_frame(video_path):
# 读取视频文件
video = cv2.VideoCapture(video_path)
# 确定视频的总帧数
total_frames = int(video.get(cv2.CAP_PROP_FRAME_COUNT))
# 设置视频的当前帧为最后一帧
video.set(cv2.CAP_PROP_POS_FRAMES, total_frames - 1)
# 读取最后一帧
ret, frame = video.read()
# 关闭视频文件
video.release()
return frame
def save_frame_as_image(video_path, output_dir):
# 提取文件名(不包含扩展名)
filename = os.path.splitext(os.path.basename(video_path))[0]
# 提取最后一帧
last_frame = extract_last_frame(video_path)
# 构造输出图片路径
output_path = os.path.join(output_dir, filename + '.jpg')
# 保存最后一帧为图片
cv2.imwrite(output_path, last_frame)
print(f"已保存最后一帧为图片:{output_path}")
# 设置视频文件夹路径
video_folder = r'C:\Users\15840\Desktop\C3\C3-1'
# 设置输出文件夹路径
output_folder = r'C:\Users\15840\Desktop\C3\picture'
# 遍历视频文件夹中的所有文件
for filename in os.listdir(video_folder):
# 构造视频文件的完整路径
video_path = os.path.join(video_folder, filename)
# 检查文件是否为视频文件
if os.path.isfile(video_path) and filename.lower().endswith(('.mp4', '.avi', '.mkv')):
# 保存最后一帧为图片
save_frame_as_image(video_path, output_folder)
用到了opencv模块,所以要确保提前进行了安装。
只需要修改视屏的文件夹路径以及需要图片的保存路径即可。