Journal
JOURNAL OF MARINE SCIENCE AND ENGINEERING
Volume 10, Issue 3, Pages -Publisher
MDPI
DOI: 10.3390/jmse10030377
Keywords
object detection; maritime dataset; deep learning; data relabel
Categories
Funding
- Dongguk University Research Fund
- Future Challenge Program through the Agency for Defense Development - Defense Acquisition Program Administration
Ask authors/readers for more resources
SMD-Plus is an improved version of SMD dataset with corrected annotations and addresses the issues of noisy labels and imprecisely located bounding boxes. It utilizes an online transformation technique called Copy & Paste to solve the class-imbalance problem in the training dataset and applies mix-up technique for augmentation in YOLO-V5 model. Experimental results show improved detection and classification performance of modified YOLO-V5 using SMD-Plus.
SMD (Singapore Maritime Dataset) is a public dataset with annotated videos, and it is almost unique in the training of deep neural networks (DNN) for the recognition of maritime objects. However, there are noisy labels and imprecisely located bounding boxes in the ground truth of the SMD. In this paper, for the benchmark of DNN algorithms, we correct the annotations of the SMD dataset and present an improved version, which we coined SMD-Plus. We also propose augmentation techniques designed especially for the SMD-Plus. More specifically, an online transformation of training images via Copy & Paste is applied to solve the class-imbalance problem in the training dataset. Furthermore, the mix-up technique is adopted in addition to the basic augmentation techniques for YOLO-V5. Experimental results show that the detection and classification performance of the modified YOLO-V5 with the SMD-Plus has improved in comparison to the original YOLO-V5. The ground truth of the SMD-Plus and our experimental results are available for download.
Authors
I am an author on this paper
Click your name to claim this paper and add it to your profile.
Reviews
Recommended
No Data Available