【OpenCV+pyqt5】视频抽帧裁剪与图片转视频
目录
【OpenCV+pyqt5】视频抽帧裁剪与图片转视频
- 本文利用OpenCV对视频进行读取，并进行抽帧，可指定时间段和抽帧间隔
- 对视频进行裁剪，裁剪设定时间段内的视频
- 对指定文件夹下的图像进行视频转换
pyqt5搭建界面
界面功能简介
- 界面比较简单，左侧显示视频，右侧提供操作按钮
- 视频下方有进度条（暂时不能调整进度），和后面的时间
- 右侧操作区有三个功能切换，右下方显示操作进度条
功能测试
- 视频抽帧有全部抽帧和时间段抽帧
- 视频裁剪根据时间段进行裁剪
- 图片转换成视频，遍历文件夹下的所有图片，根据帧率合成指定视频名称
- 以上三种功能测试均通过
OpenCV功能详解
读取视频并显示视频信息
- 利用
cap = cv2.VideoCapture(video_name)进行指定视频文件的读取
def _open_video_(self):
self.video_name = os.path.split(self.video_path_name)[-1]
if self.video_name.split('.')[-1]not in ['mp4','avi','h264']:
loggingdata('video name error, {}'.format(self.video_path_name))
return -1
self.frame_now = 0
self.cap = cv2.VideoCapture(self.video_path_name)
self.frame_num = int(self.cap.get(7))
self.FPS = int(self.cap.get(5))
self.frame_w = int(self.cap.get(3))
self.frame_h = int(self.cap.get(4))
- 通过以下命令获取视频信息
cap.get(0) CV_CAP_PROP_POS_MSEC 视频文件的当前位置（播放）以毫秒为单位
cap.get(1) CV_CAP_PROP_POS_FRAMES 基于以0开始的被捕获或解码的帧索引
cap.get(2) CV_CAP_PROP_POS_AVI_RATIO 视频文件的相对位置（播放）：0 = 电影开始，1 = 影片的结尾。
cap.get(3) CV_CAP_PROP_FRAME_WIDTH 在视频流的帧的宽度
cap.get(4) CV_CAP_PROP_FRAME_HEIGHT 在视频流的帧的高度
cap.get(5) CV_CAP_PROP_FPS 帧速率
cap.get(6) CV_CAP_PROP_FOURCC 编解码的4字-字符代码
cap.get(7) CV_CAP_PROP_FRAME_COUNT 视频文件中的帧数
时间转换函数
- 通过界面输入时间
00:00:15转换成秒,【注】这里默认输入多组时间段
def _time2second(self,start_end_time):
将输入的时间转换成秒 ['00:10:40','00:10:47'] - > [600,900]
second_time = []
for span_time in start_end_time:
start_time = span_time[0].split(':')
print(start_time)
start_time = int(start_time[0])*3600 + int(start_time[1])*60 + int(start_time[2])
end_time = span_time[1].split(':')
end_time = int(end_time[0])*3600 + int(end_time[1])*60 + int(end_time[2])
# print(end_time)
if end_time < start_time:
return -1
second_time.append([start_time,end_time])
return second_time
根据获得的视频进行抽帧
- 抽帧的逻辑是：循环读取视频，当前帧如果在指定范围内，则进行保存，超过范围则退出；显然这种方式再截取较长视频的尾部会很慢。
- 另外一种逻辑就是利用
cap.set(cv2.CAP_PROP_POS_FRAMES,start_frame)指定读取位置，这种方式较快 - 但实验发现
.h264利用第二种不能实现定位，每次还是从视频起点开始读，所以本例采用前者进行抽帧；
def video2frame(self,method = 0,frame_span = 1,ex_frame_time = [0,10]):
# 1. 视频抽帧
# method = 0 整体抽帧 method = 1 时间段抽帧
loggingdata('---------------start video2frame---------------------')
self._open_video_()
ex_frame_time = [x*self.FPS for x in ex_frame_time] if method == 1 else [0,self.frame_num]
while True:
if method == 1 and self.frame_now >= ex_frame_time[1]:
loggingdata('video2frame save picture number is {},save frame form {} to {}'.format(
ex_frame_time[1],ex_frame_time[0],self.frame_now))
break
success, origin_img = self.cap.read()
if not success or len(origin_img) < 2:
break
self.frame_now +=1
if self.frame_now%frame_span != 0:
continue
if self.frame_now < ex_frame_time[1] and self.frame_now >= ex_frame_time[0]:
cv2.imwrite(os.path.join(self.img_save_path,self.video_name.split('.')[0])+"_%.6d.jpg"%self.frame_now,origin_img)
loggingdata('==============save image stop==================')
视频裁剪
- 裁剪逻辑：继承抽帧逻辑，将符合裁剪区域内的图像写入
cut_video中，帧率默认原视频，名称在原名称上增加裁剪起始帧
def cut_video(self,save_fps = 20,cut_frame_time = [0,10]):
self._open_video_()
save_fps = self.FPS
cut_frame_time = [x*self.FPS for x in cut_frame_time]
cut_video = cv2.VideoWriter(os.path.join(self.cut_video_save_path,self.video_name.split('.')[0])+\
'_'+str(cut_frame_time[0]) + '.mp4', cv2.VideoWriter_fourcc('M','P','E','G'), save_fps, (self.frame_w,self.frame_h))
while self.frame_now < cut_frame_time[1]:
success, origin_img = self.cap.read()
if not success or len(origin_img) < 2 or self.frame_now >= cut_frame_time[1]:
break
self.frame_now +=1
# print(self.frame_now)
if self.frame_now > cut_frame_time[0] and self.frame_now < cut_frame_time[1]:
cut_video.write(origin_img)
print('cut end')
cut_video.release()
图片转视频
- 和视频裁剪类似，只是将前半段的视频读取替换成图片
def img2video(self,save_fps = 20):
img_h = 0
img_w = 0
for img_name in os.listdir(self.img_files):
if img_name.split('.')[-1] not in ['bmp','jpg','jpeg','png']:
continue
tmp_img = cv2.imread(os.path.join(self.img_files,img_name))
img_h = tmp_img.shape[0]
img_w = tmp_img.shape[1]
break
img2video = cv2.VideoWriter(self.video_save_path, cv2.VideoWriter_fourcc('M','P','E','G'), save_fps, (img_w,img_h))
for img_name in os.listdir(self.img_files):
if img_name.split('.')[-1] not in ['bmp','jpg','jpeg','png']:
continue
tmp_img = cv2.imread(os.path.join(self.img_files,img_name))
img2video.write(tmp_img)
img2video.release()
20210413更新
- 1.播放条可以拖动，调整播放位置，增加视频信息展示
- 2.抽帧逻辑优化，大视频可以快速抽帧
- 3.增加抽帧数量的选项和保存图片类型
- 4.优化界面，增加图标
可执行文件下载
https://download.csdn.net/download/wangxiaobei2017/16651184
20210519更新
- 增加视频合并功能，选择视频文件夹，对所有视频进行统一读取并整合成一个视频；
- 视频裁剪增加倍速功能
视频合并
- 考虑到对较长视频进行裁剪后需要重新合并成一段视频，于是增加该功能，主要逻辑就是依次读取文件夹内的视频，保存到新视频内，可定义新视频的帧率，新视频的分辨率与文件夹内最后视频保持一致；详细代码如下：
def video_link(self,save_fps = 20,video_w=1280,video_h=1024):
if len(self.video_file_name) < 1:
loggingdata("video_link ERROR : no video file!!!")
return -1
# 多段视频拼接在一起
self.frame_now = 0
tmp_video_link = cv2.VideoWriter(self.video_save_path, cv2.VideoWriter_fourcc('M','P','E','G'), save_fps, (video_w,video_h))
for video_name in os.listdir(self.video_file_name):
if video_name.split('.')[-1]not in ['mp4','avi','h264']:
loggingdata("can not open this video {}".format(video_name))
continue
cap = cv2.VideoCapture(os.path.join(self.video_file_name,video_name))
# print(os.path.join(self.video_file_name,video_name))
while self.stop_flag:
success, origin_img = cap.read()
if not success or len(origin_img) < 2:
break
# print(self.frame_now)
self.frame_now += 1
tmp_video_link.write(cv2.resize(origin_img,(video_w,video_h)))
tmp_video_link.release()
裁剪视频增加倍速
- 裁剪视频过程中会遇到需要快放和慢放，快放的逻辑就是剔除多余帧，慢放就是增加重复帧，具体代码如下：
while self.stop_flag and self.frame_now < cut_frame_time[1]:
success, origin_img = self.cap.read()
if not success or len(origin_img) < 2 or self.frame_now > cut_frame_time[1]:
break
self.frame_now +=1
if self.frame_now > cut_frame_time[0] and self.frame_now < cut_frame_time[1]:
if video_speed >= 1:
if self.frame_now%int(video_speed) == 0:
cut_video.write(origin_img)
else:
for i in range(int(1/video_speed)):
cut_video.write(origin_img)
可执行文件下载 版本：20210519
https://download.csdn.net/download/wangxiaobei2017/18862562
posted on 2021-05-19 13:26 wangxiaobei2019 阅读(730) 评论(0) 编辑 收藏 举报
