Powered by OpenAIRE graph
Found an issue? Give us feedback
addClaim

Object detection and ship classification using YOLOv5

Authors: Sean Brown; Caitlin Hall; Raffaele Galliera; Sikha Bagui;

Object detection and ship classification using YOLOv5

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.

Related Organizations
  • BIP!
    Impact byBIP!
    selected citations
    These citations are derived from selected sources.
    This is an alternative to the "Influence" indicator, which also reflects the overall/total impact of an article in the research community at large, based on the underlying citation network (diachronically).
    2
    popularity
    This indicator reflects the "current" impact/attention (the "hype") of an article in the research community at large, based on the underlying citation network.
    Average
    influence
    This indicator reflects the overall/total impact of an article in the research community at large, based on the underlying citation network (diachronically).
    Average
    impulse
    This indicator reflects the initial momentum of an article directly after its publication, based on the underlying citation network.
    Average
Powered by OpenAIRE graph
Found an issue? Give us feedback
selected citations
These citations are derived from selected sources.
This is an alternative to the "Influence" indicator, which also reflects the overall/total impact of an article in the research community at large, based on the underlying citation network (diachronically).
BIP!Citations provided by BIP!
popularity
This indicator reflects the "current" impact/attention (the "hype") of an article in the research community at large, based on the underlying citation network.
BIP!Popularity provided by BIP!
influence
This indicator reflects the overall/total impact of an article in the research community at large, based on the underlying citation network (diachronically).
BIP!Influence provided by BIP!
impulse
This indicator reflects the initial momentum of an article directly after its publication, based on the underlying citation network.
BIP!Impulse provided by BIP!
2
Average
Average
Average
Upload OA version
Are you the author of this publication? Upload your Open Access version to Zenodo!
It’s fast and easy, just two clicks!