BAMBI Dataset
收藏BAMBI 数据集概述
数据集简介
BAMBI 数据集是一个大规模机载多光谱野生动物数据集,包含 389 个配对的 RGB 和热成像航空视频序列,记录于奥地利多样化的森林及森林邻近栖息地。每个帧都带有精确的全球坐标(经度、纬度和海拔)地理参考,支持在图像空间和地理空间中进行学习和评估。
数据集概览
| 属性 | 数值 |
|---|---|
| 视频序列 | 389 个配对的 RGB + 热成像 |
| 总飞行时间 | ~45 小时 |
| 标注轨迹 | 5,100 条 |
| 关键帧 | 92,701 帧 |
| 插值边界框 | 1,218,903 个 |
| 物种类别 | 12 类 |
| 记录时间 | 2023年1月 – 2024年11月 |
| 记录地点 | 奥地利(蒂罗尔州、上奥地利州、下奥地利州、萨尔茨堡州、施蒂利亚州、克恩顿州) |
物种覆盖
数据集涵盖以下 12 个类别:
| # | 物种 | 通用名 | Wikidata | 轨迹数 | 关键帧数 |
|---|---|---|---|---|---|
| 1 | Sus scrofa | 野猪 | Q58697 | 1,770 | 26,132 |
| 2 | Cervus elaphus | 马鹿 | Q1219579 | 1,613 | 26,674 |
| 3 | Capreolus capreolus | 狍 | Q122069 | 682 | 9,404 |
| 4 | Dama dama | 黇鹿 | Q20908334 | 297 | 20,003 |
| 5 | Capra ibex | 阿尔卑斯野山羊 | Q168327 | 100 | 3,005 |
| 6 | Rupicapra rupicapra | 岩羚羊 | Q131340 | 15 | 747 |
| 7 | Aves | 鸟类 | Q5113 | 75 | 942 |
| 8 | Homo sapiens | 人类 | Q15978631 | 93 | 1,158 |
| 9 | Canis lupus familiaris | 狗 | Q26972265 | 7 | 95 |
| 10 | Sus scrofa × Sus domesticus | 杂交猪 | Q602666 | 44 | 1,484 |
| 11 | — | 非动物 | Q10738 | 60 | 521 |
| 12 | — | 未知 | Q24238356 | 344 | 2,536 |
标注格式
标注以 CSV 文件(无表头)形式存储,遵循自定义的多目标跟踪格式:
frame, track_id, bb_left, bb_top, bb_width, bb_height, conf, class_id, visibility, species, gender, age, is_propagated
| 列名 | 类型 | 描述 |
|---|---|---|
frame |
int | 帧索引 |
track_id |
int | 唯一轨迹标识符 |
bb_left |
int | 边界框左坐标(像素) |
bb_top |
int | 边界框顶坐标(像素) |
bb_width |
int | 边界框宽度(像素) |
bb_height |
int | 边界框高度(像素) |
conf |
float | 置信度分数 |
class_id |
int | 物种类别标识符 |
visibility |
float | 可见度(1.0 = 完全可见,0.0 = 完全遮挡) |
species |
str | 物种名称 |
gender |
int | 性别(0 = 未知,1 = 雄性,2 = 雌性) |
age |
int | 年龄(0 = 未知,1 = 幼年,2 = 成年) |
is_propagated |
int | 0 = 标注的关键帧,1 = 插值帧 |
类别标识符定义为 <wikidata_id>-<gender>-<age>-<visibility>,具体映射关系详见 README 中的 JSON 结构。
数据可用性
数据集在 Zenodo 上公开可用:
- https://doi.org/10.5281/zenodo.18692354
- https://doi.org/10.5281/zenodo.18698508
- https://doi.org/10.5281/zenodo.18703312
- https://doi.org/10.5281/zenodo.18705705
- https://doi.org/10.5281/zenodo.18707610
- https://doi.org/10.5281/zenodo.18711217
- https://doi.org/10.5281/zenodo.18715162
- https://doi.org/10.5281/zenodo.18717601
数据处理脚本
该仓库提供了用于下载、处理和可视化数据集的示例脚本。主要功能包括:
- 自动下载:从 Zenodo 选择性下载飞行数据 ZIP 文件。
- 飞行筛选:根据物种、遮挡、性别、年龄、天气、日期范围和无人机名称筛选飞行数据。
- 帧提取:从并排视频文件中提取热成像和 RGB 帧。
- MOT 插值:使用边界框坐标的线性插值在标注的关键帧之间填充缺失帧。
- MOT 筛选:按物种、类别 ID、可见度、边界框大小、性别或年龄筛选标注文件。
- MOT 到 YOLO 转换:将 MOT 标注转换为 YOLO 标签格式。
- 可视化:在提取的帧或视频上叠加边界框。
- 从位姿生成 DEM:根据位姿文件中的 GPS 坐标下载、合并和裁剪数字高程模型数据。
- 在位姿中添加相对 DEM 位置:基于 DEM 元数据将相对位置偏移添加到无人机位姿文件。
RGB-热成像帧匹配说明
尽管无人机同时记录 RGB 和热成像图像,但两种模态在时间上并不同步。时间偏移随飞行动态变化,因此热成像帧的标注不能直接转移到 RGB 图像——必须在序列或帧级别建立对应关系。数据集包含一种基于局部块匹配的策略来对齐热成像标注与其 RGB 对应部分。
相关资源
- BAMBI BBox Corrections:https://github.com/HugoMarkoff/BAMBI_BBox_Corrections
- AlfsPY:https://github.com/bambi-eco/alfs_py
- Detection:https://github.com/bambi-eco/bambi_detection
- Geo-Referenced Tracking:https://github.com/bambi-eco/Geo-Referenced-Tracking
- Bambi-QGIS:https://github.com/bambi-eco/Bambi-QGIS
引用
bibtex @inproceedings{bambi2026, title = {The BAMBI Dataset: Multimodal Nadir UAV-Recordings of Forest Wildlife}, author = {TODO}, booktitle = {TODO}, year = {2026} }
许可证
此仓库采用 MIT 许可证。



