python如何使用opencv提取光流详解
作者:bug_Cat 发布时间:2023-06-19 19:02:19
标签:python,opencv,光流
前言
光流flow特征中包含了一个视频当中运动相关的信息,在视频动作定位当中光流特征使用的比较多,所以记录一下提取光流特征的方法。
使用的方法是TVL1方法,最终提取的光流图片还可以配合I3D模型进行特征的提取。光流的计算先需要将视频一帧一帧提取出来,然后再通过连续两帧之间的差异进行计算。
提取帧
提取视频的帧的算法如下:
其中video_list.txt
中写的是视频的名字,也就是告诉程序需要将那些视频提取帧:
videos
中存放视频,与video_list.txt
中写的视频名字对应
import cv2
import numpy as np
import os
import multiprocessing
video_root = 'video_list.txt'
root = 'videos'
out_root = 'frames'
suffix = '.jpg'
def save_image(root, vid_name, num, image):
file_name = os.path.join(root, vid_name, str(num) + suffix)
# print(file_name)
cv2.imwrite(file_name, image)
def process(vid_path, preffix):
videoCapture = cv2.VideoCapture(vid_path)
i = 0
while True:
success, frame = videoCapture.read()
if success:
i = i + 1
save_image(out_root, preffix, i, frame)
# print('save image vid name: ', file_name, '; frame num: ', i)
else:
break
def main(root):
if not os.path.exists(out_root):
os.mkdir(out_root)
# path_list = os.listdir(root)
path_list = []
#### 读取txt中视频信息 ####
with open(video_root, 'r') as f:
for id, line in enumerate(f):
video_name = line.strip().split()
path_list.append(video_name[0])
pool = multiprocessing.Pool(processes=4)
for file_name in path_list:
path = os.path.join(root, file_name)
preffix = file_name.split('.')[0]
dir_name = os.path.join(out_root, preffix)
if not os.path.exists(dir_name):
os.mkdir(dir_name)
pool.apply_async(process, args=(path, preffix))
# process(path,preffix)
pool.close()
pool.join()
if __name__ == '__main__':
main(root)
print("finish!!!!!!!!!!!!!!!!!!")
运行完这个程序就能将需要提取的视频帧放在frames
对应的目录下。
提取flow光流
提取光流使用了opencv模块,主要通过上面提取的视频帧进行计算,光流计算使用cpu资源比较多,所以会计算很长时间。
光流提取的代码如下:
import cv2
import os
import numpy as np
import glob
import multiprocessing
###### 使用frames帧进行 flow光流计算
video_root = 'video_list.txt'
root = 'frames'
out_root = 'flow'
def cal_for_frames(video_path):
# print(video_path)
frames = glob.glob(os.path.join(video_path, '*.jpg'))
frames.sort()
flow = []
prev = cv2.imread(frames[0])
prev = cv2.cvtColor(prev, cv2.COLOR_BGR2GRAY)
for i, frame_curr in enumerate(frames[1:]):
curr = cv2.imread(frame_curr)
curr = cv2.cvtColor(curr, cv2.COLOR_BGR2GRAY)
tmp_flow = compute_TVL1(prev, curr)
flow.append(tmp_flow)
prev = curr
return flow
def compute_TVL1(prev, curr, bound=15):
TVL1 = cv2.optflow.DualTVL1OpticalFlow_create()
flow = TVL1.calc(prev, curr, None)
assert flow.dtype == np.float32
flow = (flow + bound) * (255.0 / (2 * bound))
flow = np.round(flow).astype(int)
flow[flow >= 255] = 255
flow[flow <= 0] = 0
return flow
def save_flow(video_flows, flow_path):
if not os.path.exists(flow_path):
os.mkdir(os.path.join(flow_path))
for i, flow in enumerate(video_flows):
cv2.imwrite(os.path.join(flow_path, str(i) + '_x.jpg'), flow[:, :, 0])
cv2.imwrite(os.path.join(flow_path, str(i) + '_y.jpg'), flow[:, :, 1])
def process(video_path, flow_path):
flow = cal_for_frames(video_path)
save_flow(flow, flow_path)
def extract_flow(root, out_root):
if not os.path.exists(out_root):
os.mkdir(out_root)
# dir_list = os.listdir(root)
dir_list = []
### 读取txt中视频信息
with open(video_root, 'r') as f:
for id, line in enumerate(f):
video_name = line.strip().split()
preffix = video_name[0].split('.')[0]
dir_list.append(preffix)
pool = multiprocessing.Pool(processes=4)
for dir_name in dir_list:
video_path = os.path.join(root, dir_name)
flow_path = os.path.join(out_root, dir_name)
# flow = cal_for_frames(video_path)
# save_flow(flow,flow_path)
# print('save flow data: ',flow_path)
# process(video_path,flow_path)
pool.apply_async(process, args=(video_path, flow_path))
pool.close()
pool.join()
if __name__ == '__main__':
extract_flow(root, out_root)
print("finish!!!!!!!!!!!!!!!!!!")
环境配置
提取光流时需要使用到cv2.optflow.DualTVL1OpticalFlow_create()
,这玩意安装有时候会有版本问题,所以安装的opencv-python和pencv-contrib-python最好版本相同
pip install opencv-python==4.1.2.30
pip install opencv-contrib-python==4.1.2.30
结果
最终flow光流图和提取的帧之间如下图所示,可以看到一些梳头发的动作变化。
来源:https://blog.csdn.net/qq_41474648/article/details/126010124
0
投稿
猜你喜欢
- 改变图像大小意味着改变尺寸,无论是单独的高或宽,还是两者。也可以按比例调整图像大小。这里将介绍resize()函数的语法及实例。语法函数原型
- 需要在两个文件中实现:首先,在talker.asp(在线名单)中做如下处理:<%p1=trim(application("v
- 本文实例讲述了django框架创建应用操作。分享给大家供大家参考,具体如下:18.1.5 安装Django安装Djangonod
- 这篇文章主要介绍了Python Django 简单分页的实现代码解析,文中通过示例代码介绍的非常详细,对大家的学习或者工作具有一定的参考学习
- 我正在开发一个档案管理系统,需要从数据库中同时调出图像及相关的文字说明,可我只做到了单纯地显示图片,像有一个数据库CHUNFENG,在数据库
- 5位数日期戳读取 .mat 文件处理里面数据时,发现里面的日期数据全部都是 “5位数” 数字,很不解;后来查到可以在excel中通过设置单元
- 有时我们有很多文件(如图片),我们需要对每一个文件进行操作。 我们还需要一份文件的名字来进行遍历,这时我们首先需要建立一份文件名单,有时还会
- 本文实例讲述了Python实现两个list求交集,并集,差集的方法。分享给大家供大家参考,具体如下:在python中,数组可以用list来表
- 1.IP代理某些网站会检测一段时间内某IP的访问次数,若访问次数过多会禁止访问,这时需要设置一些代理服务器,每隔一段时间换一个代理。IP代理
- 在python代码编写过程中,养成注释的习惯非常有用,可以让自己或别人后续在阅读代码时,轻松理解代码的含义。如果只是简单的单行注释,可直接用
- 有如下 Pandas DataFrame:import pandas as pdinp = [{'c1':10, '
- 在记忆里,关于时间方面常的SQL也就下面这两个了,大多数朋友问题中所涉及到的数据库都ACCESS的,在些,也就写出这两SQL了。年代久远,目
- python的try语句有两种风格一是处理异常(try/except/else)二是无论是否发生异常都将执行最后的代码(try/finall
- 本文为大家分享了python2.7和NLTK安装教程,具体内容如下系统:Windows 7 Ultimate 64-bitsPython 2
- 最近,在项目开发过程中,碰到了数据库死锁问题,在解决问题的过程中,笔者对MySQL InnoDB引擎锁机制的理解逐步加深。案例如下:在使用S
- 用header 发送cookie header("Set-Cookie: testcookie=中文
- 本文实例讲述了基于python编写的微博应用,分享给大家供大家参考。具体如下:在编写自己的微博应用之前,先要到weibo开放平台申请应用的公
- Python初学者小游戏:猜数字游戏逻辑:电脑随机生成一个数字,然后玩家猜数字,电脑提示猜的数字大了还是小了,供玩家缩小数字范围,达到既定次
- 最近由于公司有一个向谷歌网站上传文件的需求,需要进行web的自动化测试,选择了selenium这个自动化测试框架,以前没有接触过这门技术,所
- 制作爬虫的步骤制作一个爬虫一般分以下几个步骤:分析需求分析网页源代码,配合开发者工具编写正则表达式或者XPath表达式正式编写 python