【反无人机目标检测与跟踪】DUT Anti-UAV数据集介绍

DUT Anti-UAV数据集是IEEE TITS 2023上大连理工大学团队提出的Vision-based Anti-UAV Detection and Tracking论文中提出的一个可见光反无人机检测与跟踪数据集。

因为反无人机目标检测与跟踪的数据集都比较少,因此这个数据集也非常宝贵。

这个数据集的链接如下

数据集链接

作者给出了数据集中检测与跟踪两部分的谷歌网盘和百度网盘的链接,可以直接下载

数据集简介

其中包含检测和跟踪子集。检测数据集包括训练集(5200个图像)、验证集(2600个图像)和测试集(2200个图像)。跟踪数据集包括20个序列。

DUT反无人机数据集包含检测和跟踪子集。检测数据集被分成训练集、测试集和验证集。跟踪数据集包含20个短期和长期序列。所有帧和图像都经过精确的手动注释。图像和物体的详细信息如表I所示。具体来说,检测数据集总共包含10,000幅图像,其中训练集、测试集和验证集分别具有5200、2200和2600幅图像。考虑到一幅图像包含多个目标的情况,检测目标的总数为10,109个,其中训练集、测试集和验证集分别具有5243个、2245个和2621个目标。

数据集特点

图像分辨率:数据集包含各种分辨率的图像。对于检测数据集,最大图像的高度和宽度为3744和5616,而最小图像的大小为160 × 240;他们之间的巨大差异。跟踪数据集具有1080 × 1920和720 × 1280分辨率的两种类型帧。图像分辨率的各种设置可以使模型适应不同大小的图像,避免过拟合。

目标和背景:为了丰富物体的多样性,防止模型过度拟合,我们选择了超过35种类型的无人机。在图3中可以看到几个示例。数据集中的场景信息也是多样的。鉴于无人机大多在户外飞行,我们数据集的背景室外环境,包括天空、乌云、丛林、高层建筑、居民楼、农田、操场。此外,我们的数据集中还考虑了各种光照条件(如白天、夜晚、黎明和黄昏)和不同的天气(如晴天、阴天和下雪天)。图4中示出了来自检测子集的各种示例。数据集中复杂的背景和明显的室外照明变化对于训练鲁棒和高性能的无人机检测模型至关重要。

图3。数据集中不同类型无人机的示例

图4。我们的数据集的检测图像和注释的示例

目标比例:无人机的尺寸往往很小,户外环境广阔。因此,在我们的数据集中,小目标的比例很大。我们根据完整图像计算物体面积比,并绘制比例分布的直方图,分别如表I和图2所示。对于检测数据集,包括训练集、测试集和验证集,平均目标面积比约为0.013,最小目标面积比为1.9 E-06,最大目标占整个图像的0.7。大多数物体都很小。整个图像中的目标大小的比例大约小于0.05。对于跟踪数据集,序列中目标的比例平滑变化。平均物体面积比为0.0031,最大比为0.045,最小比为2.7 E-04。与一般检测和跟踪数据集中的目标相比,小目标更难检测和跟踪,也更容易出现故障,如漏检和跟踪丢失。

图2.DUT反无人机数据集的纵横比和比例分布

目标纵横比:表I和图2还示出了物体纵横比。我们数据集中的目标有不同的纵横比,最大值为6.67,最小值为1.0。在一个序列中,相同的物体具有显著的纵横比变化。例如,"video10"中的目标纵横比在1.0和4.33之间变化。大多数目标的纵横比在1.0到3.0之间。

目标位置:图1以散点图的形式描述了目标相对中心位置的位置分布。大多数物体都集中在图像的中心。所有集合中目标运动的范围不同,并且目标的水平和垂直运动分布均匀。对于跟踪数据集,一个序列中的目标的边界框是连续的。根据图1(d),除了图像的中心区域之外,目标还频繁地向图像的右侧和左下方移动。

图1.DUT反无人机数据集的位置分布

数据集挑战

通过对最后一小节中提出的数据集特征的分析,发现无人机检测和跟踪遇到了许多困难和挑战。主要挑战是物体太小、背景复杂或与物体相似、光线变化明显。也容易出现物体模糊、快速运动、相机运动和视野外的情况。图4和图5分别示出了反映上述挑战的检测和跟踪数据集的示例。

图4。数据集的检测图像和注释的示例

图5。数据集的跟踪序列和注释的示例

相关推荐
怪味&先森15 分钟前
利用pytorch对加噪堆叠自编码器在MNIST数据集进行训练和验证
人工智能·pytorch·python
城电科技23 分钟前
城电科技 | 太阳能花怎么选择?光伏太阳花的应用场景在哪里?
人工智能·科技·能源
MPCTHU1 小时前
预测分析(四):面向预测分析的神经网络简介
人工智能·深度学习·神经网络
gqkmiss1 小时前
Browser-use:基于 Python 的智能浏览器自动化 AI 工具调研与实战
人工智能·python·ai·自动化·浏览器
蚂蚁在飞-1 小时前
浏览器自动化操作AI工具-browser-use
运维·人工智能·自动化
jndingxin2 小时前
OpenCV 图形API(17)计算输入矩阵 src 中每个元素的平方根函数sqrt()
人工智能·opencv
Lilith的AI学习日记2 小时前
LangChain核心架构解析:从传统Chain到LCEL的演进之路
人工智能·架构·langchain·ai编程
qq_273900232 小时前
Pytorch torch.nn.utils.rnn.pad_sequence 介绍
人工智能·pytorch·python·rnn·深度学习
大模型真好玩2 小时前
点名DeepSeek?一文解析”最强大模型“Llama-4到底是王者归来 还是 困兽犹斗?
人工智能·llama·deepseek