2023/05/16 by Andrew Farley, Mohsen Zand, Farley, Andrew +3 · 1 citation
Computer Science · Engineering · Social Sciences · #Artificial Intelligence (cs.AI) #Computer Vision and Pattern Recognition (cs.CV) #FOS: Computer and information sciences #Human Mobility and Location-Based Analysis #Traffic Prediction and Management Techniques #Video Surveillance and Tracking Methods
paper · pdf · doi:10.48550/arxiv.2305.09401
openalex publication_date 2023/05/16 · openalex created_date 2023/05/19 · openalex updated_date 2026/07/28
We propose a method that augments a simulated dataset using diffusion models to improve the performance of pedestrian detection in real-world data. The high cost of collecting and annotating data in the real-world has motivated the use of simulation platforms to create training datasets. While simulated data is inexpensive to collect and annotate, it unfortunately does not always closely match the distribution of real-world data, which is known as the sim2real gap. In this paper we propose a novel method of synthetic data creation meant to close the sim2real gap for the challenging pedestrian detection task. Our method uses a diffusion-based architecture to learn a real-world distribution which, once trained, is used to generate datasets. We mix this generated data with simulated data as a form of augmentation and show that training on a combination of generated and simulated data increases average precision by as much as 27.3% for pedestrian detection models in real-world data, compared against training on purely simulated data.