Object detection and ship classification using YOLOv5
-
Published:2023
Issue:1
Volume:4
Page:7-16
-
ISSN:
-
Container-title:BOHR International Journal of Smart Computing and Information Technology
-
language:
-
Short-container-title:BIJSCIT
Author:
Brown Sean, ,Hall Caitlin,Galliera Raffaele,Bagui Sikha, , ,
Abstract
Using a public dataset of images of maritime vessels provided by Analytics Vidhya, manual annotations were madeon a subsample of images with Roboflow using the ground truth classifications provided by the dataset. YOLOv5,a prominent open source family of object detection models that comes with an out-of-the-box pre-training on theCommon Objects in Context (COCO) dataset, was used to train on annotations of sub-classifications of maritimevessels. YOLOv5 provides significant results in detecting a boat. The training, validation, and test set of imagestrained YOLOv5 in the cloud using Google Colab. Three of our five subclasses, namely, cruise ships, ROROs (RollOn Roll Off, typically car carriers), and military ships, have very distinct shapes and features and yielded positiveresults. Two of our subclasses, namely, the tanker and cargo ship, have similar characteristics when the cargoship is unloaded and not carrying any cargo containers. This yielded interesting misclassifications that could beimproved in future work. Our trained model resulted in the validation metric of mean Average Precision (mAP@0.5)of 0.932 across all subclassification of ships.
Publisher
BOHR Publishers
Subject
General Medicine,Fluid Flow and Transfer Processes,Building and Construction,General Medicine,General Engineering,General Earth and Planetary Sciences,General Engineering,General Environmental Science,General Earth and Planetary Sciences,General Environmental Science,General Medicine,General Earth and Planetary Sciences,General Environmental Science,General Medicine,General Medicine
Cited by
1 articles.
订阅此论文施引文献
订阅此论文施引文献,注册后可以免费订阅5篇论文的施引文献,订阅后可以查看论文全部施引文献
1. Spring Rolls' Size Inspection System Using Deep Image Processing;2024 21st International Conference on Electrical Engineering/Electronics, Computer, Telecommunications and Information Technology (ECTI-CON);2024-05-27