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.
| Model | Bulk cargo carrier | Container ship | Fishing boat | General cargo ship | Ore carrier | Passenger ship | |
| YOLOv4 | 92.79% | 98.61% | 94.21% | 93.83% | 94.29% | 90.08% | 93.97% | YOLOv5 | 95.80% | 99.50% | 95.90% | 98.50% | 95.00% | 96.00% | 96.80% | YOLOv7 | 96.89% | 99.99% | 95.49% | 97.48% | 96.66% | 93.33% | 96.64% | YOLOv7-RCBAM | 96.90% | 100% | 95.93% | 98.16% | 97.99% | 96.56% | 97.59% |
|
|