MetDetPy 是一个基于 python 开发的,可从直录视频或图像中检测流星的检测器。其视频检测受到uzanka/MeteorDetector项目的启发。MetDetPy具有以下优点:
-
易于使用且可配置: MetDetPy设计有易用的默认配置,多数情况下无需详细配置参数即可以进行高效的流星检测,并且也支持修改设置以获得更好的检测结果。
-
适用于各种设备和曝光时间: MetDetPy可以从各种设备拍摄的视频及图像中检测流星。借助一系列自适应算法与深度学习模型,MetDetPy对从流星监控到数码相机拍摄的数据都能较好工作。
-
低 CPU 使用率: MetDetPy 工作时对 CPU 和内存占用率较低,可支持在主流电脑或准系统上进行多摄像头输入的实时检测。
-
可选的深度学习模型接入: MetDetPy 已接入深度学习支持,可以在主检测或重校验阶段选择性使用深度学习模型,在不显著增加性能开销的情况下提升检测效果。模型也可用于图像中的流星检测。
-
有效的过滤器: 流星结果将根据其视觉特性与运动属性进行重校验以排除误报样本。每个预测都将被给出一个取值范围为
[0,1]
的置信度分数,表示其被认为是流星的可能性。 -
丰富的支持工具: MetDetPy 还提供了数个工具以支持评估和导出功能,包括一个评估工具和剪辑工具,以支持进行高效的视频切片、图像堆叠或结果评估。
你可以从 Release 处获取最新的MetDetPy发行版。发行版将 MetDetPy 进行了打包,可独立在主流平台运行(Windows,macOS及Linux)。你也可以自行使用 nuitka
或 pyinstaller
构建独立的可执行文件(见 打包Python代码为可执行程序)。
此外,MetDetPy 从 Meteor Master 的 1.2.0 版本开始作为其后端。Meteor Master (AI)是由 奔跑的龟斯 开发的流星检测软件,在MetDetPy的基础上提供了完善的GUI,多种直播流支持,便捷的导出和自动启停等功能。可以从 Meteor Master官方网站 获取更多信息,或从微软商店/App Store获取其最新版。其早期版本可从 百度网盘 (Access Code: jz01) 获取。
- 64bit OS
- Python>=3.7 (推荐 3.9+)
- numpy>=1.15.0
- opencv_python>=4.9.0
- tqdm>=4.0.0
- easydict>=1.0
- multiprocess>=0.70.0
- onnxruntime>=1.16.0
可以通过如下指令安装依赖:
pip install -r requirements.txt
上述软件包能够使 MetDetPy 正常运行,但深度学习模型仅支持在 CPU 设备上运行。如果希望利用 GPU,可以按照以下方式额外安装或替换 onnxruntime 相关库:
-
Windows/Linux 用户(推荐): 如果您使用的是 Windows 或 Linux,建议额外安装
onnxruntime_directml
。该库利用 DirectX 进行模型推理加速,适用于大多数 GPU(包括 Nvidia、AMD、Intel 等厂商的显卡)。 -
Nvidia GPU 用户(高级): 如果您使用的是 Nvidia GPU 并且已安装 CUDA,可以安装与 CUDA 版本匹配的
onnxruntime-gpu
代替onnxruntime
。这可以启用 CUDA 加速,从而带来更高的性能。
-
对于 macOs 用户,CoreML模型推理加速已经被集成到
onnxruntime
中,无需额外配置即可启用GPU支持。 -
在当前发布版本中,Windows 软件包使用
onnxruntime_directml
。默认的 CUDA 支持将在准备好后添加。
MetDetPy
是视频流星检测的启动器,其用法如下:
python MetDetPy.py target [--cfg CFG] [--mask MASK] [--start-time START_TIME] [--end-time END_TIME]
[--exp-time EXP_TIME] [--mode {backend,frontend}] [--debug]
[--resize RESIZE] [--adaptive-thre ADAPTIVE_THRE] [--bi-thre BI_THRE | --sensitivity SENSITIVITY]
[--recheck RECHECK] [--save-rechecked-img SAVE_RECHECKED_IMG]
[--provider {cpu,default,coreml,dml,cuda}][--live-mode {on,off}][--save SAVE]
-
target
: 待检测视频文件。支持常见的视频编码。 -
--cfg
: 配置文件。默认情况下使用config目录下的m3det_normal.json文件。 -
--mask
:指定掩模(遮罩)图像。可以使用任何非白色颜色的颜色在空白图像上覆盖不需要检测的区域来创建掩馍图像。不强制要求尺寸与原图相同。支持JPEG和PNG格式。 -
--start-time
:检测的开始时间。可以输入单位为ms的整数或是形如"HH:MM:SS"
的字符串。默认从头开始分析。 -
--end-time
:检测的结束时间。可以输入单位为ms的整数或是形如"HH:MM:SS"
的字符串。不指定将分析到视频结尾。 -
--mode
:运行模式。从{backend, frontend}
中选择。frontend
运行时会显示运行相关信息的进度条,backend 则具有随时刷新的输出流,适合作为后端时使用管道进行输出。默认情况下使用frontend
。 -
--debug
: 调试模式。以调试模式启动 MetDetPy 时,会打印更详细的日志信息用于调试。 -
--visu
: 可视模式。以可视模式启动时,会创建一个额外视频窗口,显示当前的检测情况。 -
--live-mode
: 实时模式。当在实时模式下运行时,检测的运行时间将会被控制在尽可能接近实际视频时长的程度,这可以帮助均衡CPU的开销。 从{on, off}
中选择。 -
--provider
: 指定优先使用的模型后端,可选值会根据平台有所差异。如果指定的模型后端不可用,则会使用默认选项。 -
--save-path
: 保存MDRF格式的检测结果到给定的路径或JSON文件下。
以下参数在不设置时使用配置文件中的默认值,如果设置则覆盖配置文件中的数值。有关这些参数的详细解释可以参考配置文件文档。
-
--resize
: 检测时采用的帧图像尺寸。可以指定单个整数(如960
,代表长边长度),列表(如[960,540]
)或者字符串(如960x540
)。 -
--exp-time
: 单帧曝光时间。可用一个浮点数或从{auto, real-time, slow}
中选择一项以指定。大多数情况下可以使用"auto"
。 -
--adaptive-thre
: 检测器中是否启用自适应二值化阈值。从{on, off}中选择。 -
--bi-thre
: 指定检测器中使用的二值化阈值。当启用自适应二值化阈值时,该选项无效化。不能与--sensitivity
同时设置。 -
--sensitivity
: 检测器的灵敏度。从{low, normal, high}
中选择。当启用自适应二值化阈值时,灵敏度选项仍起效,且更高的灵敏度将会估计更高的阈值。不能与--bi-thre
同时设置。 -
--recheck
: 启用重校验机制减少误报。从{on, off}
中选择。
python MetDetPy.py "./test/20220413Red.mp4" --mask "./test/mask-east.jpg" --visu --save-path .
大多数与检测相关的重要参数都预先定义并储存在配置文件中。大多数情况下,这些预设值都能够较好的工作,但有时对参数微调可以取得更好的结果。如果想要获取配置文件的说明,可以参考配置文件文档获取更多信息。
MetDetPhoto
是图像流星检测的启动器,其用法如下:
python MetDetPhoto.py target [--mask MASK]
[--model-path MODEL_PATH] [--model-type MODEL_TYPE]
[--exclude-noise] [--debayer] [--debayer-pattern DEBAYER_PATTERN]
[--visu] [--visu-resolution VISU_RESOLUTION]
[--save-path SAVE_PATH]
-
target
: 目标图像文件/文件夹。支持单张图像,图像文件夹以及常规视频编码格式的延时视频文件。 -
--mask
: 掩模图像。可以使用任何非白色颜色的颜色在空白图像上覆盖不需要检测的区域来创建掩馍图像。不强制要求尺寸与原图相同。支持JPEG和PNG格式。 -
--model-path
: 模型权重文件路径。默认使用./weights/yolov5s_v2.onnx权重。 -
--model-type
: 模型格式,决定如何处理模型的输出。目前仅支持YOLO
格式。默认值为YOLO
。 -
--exclude-noise
: 输出时从预测排除常见噪声类型(如卫星和飞虫),仅保存正样本到文件。 -
--debayer
: 是否在处理延时视频前对视频帧进行Debayer变换。 -
--debayer-pattern
: Debayer使用的矩阵,如 RGGB 或 BGGR。仅在--debayer
选项启用时生效。 -
--visu
: 可视模式。以可视模式启动时,会创建一个额外视频窗口,显示当前的检测情况。 -
--visu-resolution
: 可视化窗口的分辨率设置。 -
--save-path
: 保存检测结果到 MDRF 格式文件中。
python MetDetPhoto.py "/path/to/your/folder" --mask "/path/to/your/mask.jpg" --exclude-noise --save-path .
MetDetPy还提供了数个工具以支持评估和导出功能,包括Evaluate(效果评估和回归测试工具)、ClipToolkit (批图像堆栈和视频剪辑工具)和make_package(打包脚本)。访问工具文档了解这些工具的使用方法。
-
在 3840x2160 10fps 视频上应用默认配置进行检测时,MetDetPy 检测流星的平均时间开销为视频长度的 20-30%(使用 Intel i5-7500 测试)。 FPS 较高的视频可能会花费更多时间。
-
我们使用从各种设备(从改装监控摄像头到数码相机)拍摄的样本视频测试 MetDetPy,MetDetPy 平均能够达到 80% 以上的准确率和 80% 以上的召回率。
-
MetDetPy 现在可以快速高效地检测大多数流星视频。 但当面对复杂的天气或其他影响因素时,其准确率和召回率还有待提高。 如果您发现 MetDetPy 在您的视频上表现不够好,欢迎联系我们或提交问题(如果可以的话,一并提供完整或剪辑的视频)。
该项目根据 Mozilla 公共许可证 2.0 (MPL-2.0) 获得许可。这意味着您可以自由使用、修改和分发该软件,但须满足以下条件:
- 源代码可用性:您对源代码所做的任何修改也必须在 MPL-2.0 许可证下可用。这确保了社区可以从改进和变化中受益。
- 文件级 Copyleft:您可以将此软件与不同许可证下的其他代码结合使用,但对 MPL-2.0 许可文件的任何修改都必须保留在同一许可证下。
- 无保证:软件按“原样”提供,不提供任何形式的明示或暗示的保证。使用风险自负。
欲了解更多详细信息,请参阅MPL-2.0许可证。
uzanka [Github]
奔跑的龟斯 [Personal Website] [Weibo] [Bilibili]
纸片儿 [Github]
DustYe夜尘 [Bilibili]
RoyalK [Weibo] [Bilibili]
MG_Raiden扬 [Weibo] [Bilibili]
星北之羽 [Bilibili]
LittleQ
韩雅南
来自偶然
杨雳鹏
Jeff戴建峰 [Weibo] [Bilibili]
贾昊
见 更新日志。