A Method to Automatic Create Dataset for Training Object Detection Neural Networks

Numerous high-accuracy neural networks for object detection have been proposed in recent years. Creating a specific training dataset seems to be the main obstacle in putting them into practice. This paper aims to overcome this obstacle and proposes an automatic dataset creation method. This method f...

Full description

Bibliographic Details
Main Authors: Shi Zhou, Zijun Yang, Miaomiao Zhu, He L Li, Seiichi Serikawa, Mitsunori Mizumachi, Lifeng Zhang
Format: Article
Language:English
Published: IEEE 2022-01-01
Series:IEEE Access
Subjects:
Online Access:https://ieeexplore.ieee.org/document/9846997/
Description
Summary:Numerous high-accuracy neural networks for object detection have been proposed in recent years. Creating a specific training dataset seems to be the main obstacle in putting them into practice. This paper aims to overcome this obstacle and proposes an automatic dataset creation method. This method first extracts objects from the source images and then combines them as synthetic images. These synthetic images are annotated automatically using the data flow in the process and can be used directly for training. To the best of our knowledge, this is the first automatic dataset creation method for object detection tasks. In addition, the adaptive object extraction method and created natural synthetic images make the proposed method maintain strong adaptation and generalization ability. To validate the feasibility, a dataset that includes 44 categories of objects is created for the object detection task in a vending supermarket. Under the strictest metric <inline-formula> <tex-math notation="LaTeX">$AP_{75} $ </tex-math></inline-formula>, both the trained EfficientDet and YOLOv4 achieve higher than 95&#x0025; in accuracy on the common difficulty testing set and higher than 90&#x0025; in accuracy on the high difficulty testing set.
ISSN:2169-3536