mmaction2——tools文件夹下
build_rawframes.py
- 用法示例
python tools/data/build_rawframes.py data/videos data/frames --task rgb --level 2 --ext mp4 --use-opencv --num-worker 8
总结：
-
只需要 RGB 帧，推荐 --use-opencv，简单高效，无需额外依赖。
-
需要光流（flow）或同时提取 RGB+光流，需安装 denseflow，不用 --use-opencv。
注意： -
–use-opencv 只支持 RGB 帧提取，不支持光流。
-
denseflow 支持光流和 RGB，但需额外安装和配置。
| 参数名 | 类型 | 默认值 | 说明 |
|---|---|---|---|
| src_dir | str | 必填 | 源视频目录 |
| out_dir | str | 必填 | 输出帧目录 |
| –task | str | flow | 提取类型：rgb / flow / both |
| –level | int | 2 | 数据目录层级（1或2） |
| –num-worker | int | 8 | 并行进程数 |
| –flow-type | str | None | 光流类型：tvl1、warp_tvl1、farn、brox |
| –out-format | str | jpg | 输出图片格式：jpg、h5、png |
| –ext str | avi | 视频文件扩展名： | avi、mp4、webm |
| –mixed-ext | flag | False | 是否处理混合扩展名的视频 |
| –new-width | int | 0 | 输出图片宽度 |
| –new-height | int | 0 | 输出图片高度（0为不变） |
| –new-short | int | 0 | 输出图片短边长度（保持比例，0为不变） |
| –num-gpu | int | 8 | GPU数量（仅光流提取时用到） |
| –resume | flag | False | 是否断点续提取 |
| –use-opencv | flag | False | 是否用 OpenCV 提取 RGB 帧 |
| –input-frames | flag | False | 是否基于已提取的 RGB 帧再提取光流 |
| –report-file | str | build_report.txt | 记录已处理文件的报告文件名 |