摘要

我们提出一个全新的、具有挑战性的数据集——CPPE-5（Medical Personal Protective Equipment，医疗个人防护装备），旨在支持对医疗个人防护装备的细粒度类别识别（即下位类别分类）研究。现有主流数据集（如PASCAL VOC、ImageNet、Microsoft COCO、OpenImages等）主要聚焦于宽泛的类别层次，无法满足此类细粒度分类任务的需求。为便于在复杂场景的实际应用中使用基于本数据集训练的模型，我们的数据集主要包含在自然场景下呈现的复杂图像，每幅图像中包含多个物体，以真实、多样的上下文环境为特征。本数据集的图像采集重点在于：尽可能获取非典型（non-iconic）图像，并确保所有图像均为真实生活场景中的实拍图像，这与当前该领域其他数据集普遍存在的理想化或合成图像存在显著区别。数据集涵盖5个物体类别：防护服（coveralls）、面罩（face shields）、手套（gloves）、口罩（masks）和护目镜（goggles），每张图像均配有边界框（bounding boxes）及对应的正样本标签。我们对CPPE-5数据集进行了详尽的分析，将其与现有的主流宽泛类别数据集以及专注于个人防护装备的数据集进行对比，发现目前尚无公开可用的同类数据集。此外，我们还对基准模型与当前先进模型在边界框检测任务上的性能表现进行了评估，并对比了不同模型的复杂度。相关代码、数据集及训练好的模型均已开源，可访问 https://git.io/cppe5-dataset 获取。