video frame interpolation
时间: 2023-08-12 12:04:22 浏览: 62
视频帧插帧是一种技术,用于在两个视频帧之间生成额外的中间帧。这可以通过使用插值算法来实现,通过分析相邻帧之间的像素值和运动信息来预测中间帧的像素值。插入这些中间帧可以提高视频的帧率,使得动作更加流畅。这对于视频编辑、动画和视频游戏等领域非常有用。常用的插帧算法包括光流法、双向插值法和深度学习模型等。
相关问题
import cv2 import numpy as np # 创建混合高斯模型 fgbg = cv2.createBackgroundSubtractorMOG2(history=500, varThreshold=50, detectShadows=False) # 打开视频文件 cap = cv2.VideoCapture('t1.mp4') # 获取视频帧率、宽度和高度 fps = int(cap.get(cv2.CAP_PROP_FPS)) width = int(cap.get(cv2.CAP_PROP_FRAME_WIDTH)) height = int(cap.get(cv2.CAP_PROP_FRAME_HEIGHT)) # 创建前景视频对象 fg_out = cv2.VideoWriter('foreground_video.avi', cv2.VideoWriter_fourcc(*'XVID'), fps, (width, height)) # 初始化上一帧 prev_frame = None # 循环遍历视频帧 while True: ret, frame = cap.read() if not ret: break # 高斯模型背景减除法 fgmask = fgbg.apply(frame) # 缩放比例 scale_percent = 50 # 计算缩放后的新尺寸 width = int(frame.shape[1] * scale_percent / 100) height = int(frame.shape[0] * scale_percent / 100) dim = (width, height) # 缩放图像 frame = cv2.resize(frame, dim, interpolation=cv2.INTER_AREA) fgmask = cv2.resize(fgmask, dim, interpolation=cv2.INTER_AREA) # 形态学开运算去除噪点 kernel = cv2.getStructuringElement(cv2.MORPH_ELLIPSE, (3, 3)) opening = cv2.morphologyEx(fgmask, cv2.MORPH_OPEN, kernel) # 寻找轮廓并计算周长 contours, hierarchy = cv2.findContours(opening, cv2.RETR_EXTERNAL, cv2.CHAIN_APPROX_SIMPLE) for cnt in contours: perimeter = cv2.arcLength(cnt, True) if perimeter > 500: # 画出矩形框 x, y, w, h = cv2.boundingRect(cnt) cv2.rectangle(frame, (x, y), (x + w, y + h), (0, 255, 0), 2) # 视频稳定 if prev_frame is not None: # 计算帧间差分 diff = cv2.absdiff(frame, prev_frame) # 计算运动向量 _, motion = cv2.optflow.calcOpticalFlowFarneback(prev_frame, frame, None, 0.5, 3, 15, 3, 5, 1.2, 0) # 平移每一帧 M = np.float32([[1, 0, motion[:,:,0].mean()], [0, 1, motion[:,:,1].mean()]]) frame = cv2.warpAffine(frame, M, (frame.shape[1], frame.shape[0])) diff = cv2.warpAffine(diff, M, (diff.shape[1], diff.shape[0])) # 显示帧间差分 cv2.imshow('diff', diff) # 更新上一帧 prev_frame = frame.copy() cv2.imshow('frame', frame) cv2.imshow('fgmask', fgmask) if cv2.waitKey(1) & 0xFF == ord('q'): break # 释放对象 cap.release() fg_out.release() cv2.destroyAllWindows()改为4.5.3版本的opencv能用的程序
import cv2 import numpy as np # 创建混合高斯模型 fgbg = cv2.createBackgroundSubtractorMOG2(history=500, varThreshold=50, detectShadows=False) # 打开视频文件 cap = cv2.VideoCapture('t1.mp4') # 获取视频帧率、宽度和高度 fps = int(cap.get(cv2.CAP_PROP_FPS)) width = int(cap.get(cv2.CAP_PROP_FRAME_WIDTH)) height = int(cap.get(cv2.CAP_PROP_FRAME_HEIGHT)) # 创建前景视频对象 fourcc = cv2.VideoWriter_fourcc(*'XVID') fg_out = cv2.VideoWriter('foreground_video.avi', fourcc, fps, (width, height)) # 初始化上一帧 prev_frame = None # 循环遍历视频帧 while True: ret, frame = cap.read() if not ret: break # 高斯模型背景减除法 fgmask = fgbg.apply(frame) # 缩放比例 scale_percent = 50 # 计算缩放后的新尺寸 width = int(frame.shape[1] * scale_percent / 100) height = int(frame.shape[0] * scale_percent / 100) dim = (width, height) # 缩放图像 frame = cv2.resize(frame, dim, interpolation=cv2.INTER_AREA) fgmask = cv2.resize(fgmask, dim, interpolation=cv2.INTER_AREA) # 形态学开运算去除噪点 kernel = cv2.getStructuringElement(cv2.MORPH_ELLIPSE, (3, 3)) opening = cv2.morphologyEx(fgmask, cv2.MORPH_OPEN, kernel) # 寻找轮廓并计算周长 contours, hierarchy = cv2.findContours(opening, cv2.RETR_EXTERNAL, cv2.CHAIN_APPROX_SIMPLE) for cnt in contours: perimeter = cv2.arcLength(cnt, True) if perimeter > 500: # 画出矩形框 x, y, w, h = cv2.boundingRect(cnt) cv2.rectangle(frame, (x, y), (x + w, y + h), (0, 255, 0), 2) # 视频稳定 if prev_frame is not None: # 计算帧间差分 diff = cv2.absdiff(frame, prev_frame) # 计算运动向量 flow = cv2.calcOpticalFlowFarneback(prev_frame, frame, None, 0.5, 3, 15, 3, 5, 1.2, 0) # 平移每一帧 mean_flow = np.mean(flow, axis=(0, 1)) M = np.float32([[1, 0, mean_flow[0]], [0, 1, mean_flow[1]]]) frame = cv2.warpAffine(frame, M, (frame.shape[1], frame.shape[0])) diff = cv2.warpAffine(diff, M, (diff.shape[1], diff.shape[0])) # 显示帧间差分 cv2.imshow('diff', diff) # 更新上一帧 prev_frame = frame.copy() cv2.imshow('frame', frame) cv2.imshow('fgmask', fgmask) if cv2.waitKey(1) & 0xFF == ord('q'): break # 释放对象 cap.release() fg_out.release() cv2.destroyAllWindows()
import cv2 import os def resize_image(image_path, width, height): """调整图片大小""" img = cv2.imread(image_path, cv2.IMREAD_UNCHANGED) resized = cv2.resize(img, (width, height), interpolation=cv2.INTER_AREA) cv2.imwrite(image_path, resized) def extract_frames(video_path, target_path): """提取视频帧并保存封面图""" try: vc = cv2.VideoCapture(video_path) # 读取视频 success, frame = vc.read() # 读取当前帧,success用于判断读取是否成功 count = 0 # 初始化计数器 while success: file_name = os.path.splitext(os.path.basename(video_path))[0] + f'_{count}.jpg' frame_path = os.path.join(target_path, file_name) cv2.imwrite(frame_path, frame) # 将当前帧保存为图片到 frame_path resize_image(frame_path, 2560, 1440) # 调整图片大小 success, frame = vc.read() # 继续读取下一帧 count += 1 # 计数器加1 except Exception as e: print(f"获取视频帧失败: {e}") if __name__ == '__main__': video_folder = 'D:/path/to/videos/1/银二-主井皮带_20230523151417' # 视频文件夹目录 target_path = 'D:/path/to/frames/1' # 帧截图保存路径 if not os.path.exists(target_path): # 如果目标路径不存在原文件夹的话就创建 os.makedirs(target_path) for file_name in os.listdir(video_folder): file_path = os.path.join(video_folder, file_name) if os.path.isfile(file_path) and file_name.endswith('.mp4'): extract_frames(file_path, target_path) print("程序执行完毕!")
这是一段Python代码,主要功能是从指定的视频文件夹中提取视频帧并保存为图片。具体实现过程如下:
1. 引入cv2和os模块。
2. 定义resize_image函数,用于调整图片大小。该函数接收三个参数:图片路径、目标宽度和目标高度。
3. 定义extract_frames函数,用于提取视频帧并保存封面图。该函数接收两个参数:视频路径和目标路径。
4. 在extract_frames函数中,通过cv2.VideoCapture读取视频,并依次读取每一帧进行处理。在读取每一帧时,先生成当前帧的文件名,然后将该帧写入到指定路径下的文件中。
5. 调用resize_image函数,将图片大小调整为指定的宽度和高度。
6. 在主函数中,指定视频文件夹路径和目标路径,并循环遍历视频文件夹下的所有mp4格式的视频文件。对于每一个mp4格式的视频文件,调用extract_frames函数提取视频帧并保存为图片。
7. 输出程序执行完毕的提示信息。
需要注意的是,该代码需要安装OpenCV库才能正常运行。