Research Article

An Improved Ship Classification Method Based on YOLOv7 Model with Attention Mechanism

Table 4

The comparison of values for each model between different ship categories.

ModelBulk cargo carrierContainer shipFishing boatGeneral cargo shipOre carrierPassenger ship

YOLOv492.79%98.61%94.21%93.83%94.29%90.08%93.97%
YOLOv595.80%99.50%95.90%98.50%95.00%96.00%96.80%
YOLOv796.89%99.99%95.49%97.48%96.66%93.33%96.64%
YOLOv7-RCBAM96.90%100%95.93%98.16%97.99%96.56%97.59%