提取视频光流成帧并写入视频中

修改一下配置文件就可以运行了

配置文件
config.py
video_path = 'xxxx/dataset/data/huaping/BXDQ05-花屏-1.mp4'#要处理的视频路径
frame_path = 'xxxx/dataset/frame'#处理成帧之后保存的路径
flow_path = 'xxxx/dataset/flow'#处理成光流之后保存的路径
save_video_path = 'xxxx/fetch-flow/output/result.mp4'#最终将光流帧变成视频的路径
fps = 30
2.将视频拆分为帧
fetchFrame.py
from __future__ import print_function
import sys
import numpy as np
import os
import imageio#用于读取和写入图像和视频的库
import cv2
import config
Height = 512
Width = 512def get_frame(file_dir):with imageio.get_reader(file_dir,  'ffmpeg') as vid:# 使用imageio库打开视频文件,并创建一个读取器对象。此对象被命名为vid。'ffmpeg'是用于读取视频的编解码器。nframes = vid.get_meta_data()['nframes']#从视频元数据中获取帧数,并保存到nframes变量中。for i, frame in enumerate(vid):n_frames = iframe = cv2.resize(frame, (Width, Height), interpolation = cv2.INTER_CUBIC)imageio.imwrite(config.frame_path+'/'+str(i)+'.jpg', frame)#将调整大小后的帧保存为JPEG文件。文件的路径和名称是由前面的路径、字符串"frame_"和当前帧索引组合而成的。np.save('nframes.npy', n_frames)#将视频的总帧数保存到一个NumPy文件中。文件名是'nframes.npy',而文件中的数据是n_frames变量。
2.计算光流
fetchFlow.py
import os
import numpy as np
import cv2
from glob import glob_IMAGE_SIZE = 256# 用于计算视频中每一帧之间的光流。它首先根据视频路径获取所有以'.jpg'结尾的帧图像文件路径,并按顺序排序。然后,它使用OpenCV库读取并转换每一帧的图像,计算当前帧与前一帧之间的光流,并将光流添加到一个列表中。最后,返回包含所有光流的列表。
import os
import cv2
from glob import globdef cal_for_frames(video_path):# 获取视频路径下的所有jpg图片路径print(video_path)# 对图片路径进行排序frames = [f for f in os.listdir(video_path)]frames = sorted(frames, key=lambda x:int(x[:-4]))#字符串排序顺序会是1,10,11这样,所以要进行关键字排序    frames = [os.path.join(video_path, f) for f in frames]print(frames)flow = []prev = cv2.imread(frames[0])  # 读取第一帧图像prev = cv2.cvtColor(prev, cv2.COLOR_BGR2GRAY)  # 将第一帧图像转换为灰度图像for i, frame_curr in enumerate(frames):print(frame_curr)curr = cv2.imread(frame_curr)  # 读取当前帧图像curr = cv2.cvtColor(curr, cv2.COLOR_BGR2GRAY)  # 将当前帧图像转换为灰度图像tmp_flow = compute_TVL1(prev, curr)  # 计算当前帧和前一帧之间的光流flow.append(tmp_flow)  # 将光流添加到列表中prev = curr  # 更新前一帧图像print(i)return flow  # 返回光流列表def compute_TVL1(prev, curr, bound=15):"""计算TV-L1光流。参数:prev: numpy.ndarray, 输入图像序列中的前一帧图像。curr: numpy.ndarray, 输入图像序列中的当前帧图像。bound: int, 光流值的边界限制,默认为15。返回值:flow: numpy.ndarray, 计算得到的光流图像。"""# 创建TV-L1光流估计算法对象TVL1 = cv2.DualTVL1OpticalFlow_create()# 使用TV-L1算法计算光流flow = TVL1.calc(prev, curr, None)assert flow.dtype == np.float32# 将光流值进行缩放和取整操作flow = (flow + bound) * (255.0 / (2*bound))flow = np.round(flow).astype(int)# 将光流值限制在0-255范围内flow[flow >= 255] = 255flow[flow <= 0] = 0return flowdef save_flow(video_flows, flow_path):"""保存视频的光流图参数:video_flows:视频的光流图(三维数组)flow_path:保存光流图的路径(字符串)返回值:无"""for i, flow in enumerate(video_flows):# 保存u分量的光流图cv2.imwrite(os.path.join(flow_path.format('u'), "{:06d}.jpg".format(i)),flow[:, :, 0])# 保存v分量的光流图cv2.imwrite(os.path.join(flow_path.format('v'), "{:06d}.jpg".format(i)),flow[:, :, 1])#u代表水平分量,v代表垂直分量def extract_flow(video_path,flow_path):flow = cal_for_frames(video_path)print("finish1")save_flow(flow, flow_path)print('finish2')print('complete:' + flow_path)return
3.将帧写入视频
main.py
import imageio
import PIL.Image as Image
import numpy as np
from fetchFlow import extract_flow
from fetchFrame import get_frame
import configvideo_path = config.video_path
fps = config.fps
flow_path = config.flow_path
frame_path = config.frame_path
save_video_path = config.save_video_path#get_frame(video_path)
extract_flow(frame_path, flow_path)flow_list = [Image.open(os.path.join(flow_path,f)) for f in os.listdir(flow_path)]
print(flow_list)
with imageio.get_writer(save_video_path, fps = fps) as video:for image in flow_list:image = image.convert('RGB')image = np.array(image)video.append_data(image)

参考:提取光流

本文来自互联网用户投稿,该文观点仅代表作者本人,不代表本站立场。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如若转载,请注明出处:http://www.mzph.cn/news/208445.shtml

如若内容造成侵权/违法违规/事实不符,请联系多彩编程网进行投诉反馈email:809451989@qq.com,一经查实,立即删除!

相关文章

自动补全的 select antd react

自动补全的 select antd react 文档&#xff1a;自动补全的 select antd react.note 链接&#xff1a;http://note.youdao.com/noteshare?idf5e4a93d2b9d6be8e459edd4eb86323b&sub19796E9BC04D4ABD9ACE325FDFF59B0E 添加链接描述 import React, { useState, useRef } from…

【1day】泛微e-office OA系统xml.php 文件 SORT_ID 参数 SQL 注入漏洞学习

注:该文章来自作者日常学习笔记,请勿利用文章内的相关技术从事非法测试,如因此产生的一切不良后果与作者无关。 目录 一、漏洞描述 二、影响版本 三、资产测绘 四、漏洞复现

理解传统模式与互联网时代 消费行为模型 AIDMA , AISAS , SICAS

1 AIDMA与AISAS 消费行为模型&#xff0c;以及所诞生的IT岗位 1.1 传统市场营销消费行为模型 AIDMA模型&#xff1a;Attention&#xff08;吸引&#xff09;&#xff0c;Interest &#xff08;兴趣&#xff09;&#xff0c;Desire&#xff08;欲望&#xff09; &#xff0c;Me…

LeetCode 76. 最小覆盖子串 滑动窗口框架

双指针的特殊应用&#xff1a;滑动窗口 代码 题目链接&#xff1a;https://leetcode.cn/problems/minimum-window-substring/description/ 不说废话&#xff0c;直接贴代码&#xff1a; static string minWindow(string s, string t) {// need记录需要匹配的字符串t中每个字…

❤ Mac IDEA使用并运行项目

❤ IDEA导入项目并运行 Mac IDEA使用 (1) 仓库导入 通过获取giett仓库包的url&#xff0c;在idea中导入项目 在gitee里获取项目的ur打开idea&#xff0c;点击 File->new->Project from Version Control (2) 创建数据库ry并导入数据脚本 &#xff08;3&#xff09;修改配…

华为配置Smart Link主备备份示例

定义 Smart Link&#xff0c;又叫做备份链路。一个Smart Link由两个接口组成&#xff0c;其中一个接口作为另一个的备份。Smart Link常用于双上行组网&#xff0c;提供可靠高效的备份和快速的切换机制。 Monitor Link是一种接口联动方案&#xff0c;它通过监控设备的上行接口…

npm私有源构建项目下载依赖报错

Jenkins构建项目报错&#xff0c;依赖找不到 Error: Couldnt find any versions for "babel/helper-module-imports" that matches "^7.22.15"at MessageError.ExtendableBuiltin (/data1/jenkins/tools/jenkins.plugins.nodejs.tools.NodeJSInstallation/…

log4j(日志的配置)

日志一般配置在resources的config下面的&#xff0c;并且Util当中的initLogRecord中的initLog&#xff08;&#xff09;方法就是加载这个log4j.properties的. 首先先看log4j.properties的配置文件 log4j.rootLoggerdebug, stdout, Rlog4j.appender.stdoutorg.apache.log4j.Co…

高性能和多级高可用,云原生数据库 GaiaDB 架构设计解析

1 云原生数据库和 GaiaDB 目前&#xff0c;云原生数据库已经被各行各业大规模投入到实际生产中&#xff0c;最终的目标都是「单机 分布式一体化」。但在演进路线上&#xff0c;当前主要有两个略有不同的路径。 一种是各大公有云厂商选择的优先保证上云兼容性的路线。它基于存…

考研真题数据结构

【2021年山西大学真题】将二叉树中所有非终端结点的左右子树交换位置&#xff0c;可以得到原二叉树的 镜像二叉树&#xff0c;如图。假设二叉树的存储形式为&#xff08;lchild&#xff0c;data&#xff0c;rchild&#xff09;&#xff0c;给出求镜像二叉树的算法: &#xff0…

Sql Server Management Studio连接Mysql

目标 已知mysql连接参数&#xff08;地址和用户&#xff09;&#xff0c;期望通过Microsoft Sql Server Management Studio &#xff08;以下简称MSSSMS&#xff09;连接Mysql&#xff0c;在MSSSMS中直接查询或修改Mysql中的数据。 下载MySql Connector/ODBC并安装&#xff0c…

使用poi-tl填充word模板,并转化为pdf输出

后端 依赖 <dependency><groupId>com.deepoove</groupId><artifactId>poi-tl</artifactId><version>1.12.0</version> </dependency>Word版本 Word版本填充代码 // 培训详情HashMap<String, Object> textMap new Ha…

maven环境搭建

maven历史版本下载&#xff1a;https://archive.apache.org/dist/maven/ 新建系统变量编辑Path&#xff0c;添加bin目录mvn -v测试查看版本号conf目录下新建repository文件夹&#xff0c;作为本地仓库 settings.xml <?xml version"1.0" encoding"UTF-8&…

2312d,d语言来绑定C++和rust

原文 各编译语言相同概念 1,按可重用函数拆分代码. 2,由源码中的函数名生成的串来标识函数.如,g为void foo()生成_Z3foov的标识.此串总是是可重现的;如,Linux上的Clang和GCC都遵循ItaniumCABI约定来装饰函数名. 3,在内存中的特定位置存储该函数的所有参数,然后用调用或等效指…

gitee配置

注册配置gitee Gitee官网 进入官网之后&#xff0c;有账号直接登录&#xff0c;没有账号注册一个新的账号 下载安装git客户端 官网地址 下载完成&#xff0c;一路直接点击安装直接安装成功 检查是否安装成功 鼠标留在桌面–>右击–>出现Git GUI Here/Git Bash Her…

windows系统nodeJs报错node-sass npm ERR! command failed

报错信息 npm WARN deprecated request2.88.2: request has been deprecated, see https://github.com/request/request/issues/3142 npm WARN deprecated tar2.2.2: This version of tar is no longer supported, and will not receive security updates. Please upgrade asa…

国科大通信原理复习

CH4-信源的数字化 26. 信源编码的基本方法和分类 27. 无失真编码和有失真编码的区别 无失真编码能够完全一模一样的恢复到原信号。 有失真编码则不行。 28. 信息量和熵的定义 29. 离散信源的最大熵定理 n表示所有符号的种类&#xff0c;比如对于二进制码字&#xff0c;Rbit对…

云计算ACP认证考试题库0-100

0001.单选题:阿里云的云盾会检查通过公共互联网登录云服务器ECS的来源IP,登录方式包括SSH和远程桌面,当来自某个IP的登录请求出现多次密码错误的情况时,会发出”ECS遭遇密码暴力破解”的报警,当收到这个报警后,最安全的处理方法应该是。 A.通知自己业务平台的所有用户立即修改…

基于支持向量机SVM的新鲜度等级预测,基于自适应粒子群优化长短期神经网络的新鲜度等级预测

目录 背影 支持向量机SVM的详细原理 SVM的定义 SVM理论 粒子群算法原理 SVM应用实例,基于支持向量机SVM的新鲜度等级预测,基于自适应粒子群优化长短期神经网络的新鲜度等级预测 代码 结果分析 展望 完整代码:基于支持向量机SVM的新鲜度等级预测,基于自适应粒子群优化长短期…

SpringBoot+线程池实现高频调用http接口并多线程解析json数据

场景 SpringbootFastJson实现解析第三方http接口json数据为实体类(时间格式化转换、字段包含中文)&#xff1a; SpringbootFastJson实现解析第三方http接口json数据为实体类(时间格式化转换、字段包含中文)-CSDN博客 Java中ExecutorService线程池的使用(Runnable和Callable多…