Unit: Building Unit Detection Dataset

TL;DR

This paper introduces a comprehensive annotated dataset of building units captured by drones, utilizing multiple enhancement techniques to improve fire detection models and aid emergency operations.

Contribution

The creation of a large, synthetic drone-captured building unit dataset with diverse scenarios and enhancement methods to improve fire detection model training.

Findings

The dataset contains 1,978 images with varied scenarios.

Enhancement techniques improve the authenticity and diversity of training data.

The dataset enhances the generalization ability of fire unit detection models.

Abstract

Fire scene datasets are crucial for training robust computer vision models, particularly in tasks such as fire early warning and emergency rescue operations. However, among the currently available fire-related data, there is a significant shortage of annotated data specifically targeting building units.To tackle this issue, we introduce an annotated dataset of building units captured by drones, which incorporates multiple enhancement techniques. We construct backgrounds using real multi-story scenes, combine motion blur and brightness adjustment to enhance the authenticity of the captured images, simulate drone shooting conditions under various circumstances, and employ large models to generate fire effects at different locations.The synthetic dataset generated by this method encompasses a wide range of building scenarios, with a total of 1,978 images. This dataset can effectively improve the generalization ability of fire unit detection, providing multi-scenario and scalable data while reducing the risks and costs associated with collecting real fire data. The dataset is available at https://github.com/boilermakerr/FireUnitData.