Research Article
An Improved Ship Classification Method Based on YOLOv7 Model with Attention Mechanism
Table 2
The comparison of
in different attention mechanisms in different ship categories.
| | Attention model | Bulk cargo carrier | Container ship | Fishing boat | General cargo ship | Ore carrier | Passenger ship |
| | YOLOv7 | 96.89% | 99.99% | 95.49% | 97.48% | 96.66% | 93.33% | | YOLOv7+ECA | 96.71% | 100% | 95.84% | 97.81% | 97.67% | 95.18% | | YOLOv7+SE | 96.20% | 100% | 95.89% | 97.78% | 97.69% | 94.16% | | YOLOv7+CBAM | 96.70% | 100% | 95.75% | 97.76% | 97.96% | 93.93% | | YOLOv7-RCBAM | 96.90% | 100% | 95.93% | 98.16% | 97.99% | 96.56% |
|
|
