Research Article
An Improved Ship Classification Method Based on YOLOv7 Model with Attention Mechanism
Table 1
The diagram of YOLOv7-RCBAM.
| | Type/stride | Filter shape | Stride | Input size | Output size |
| | Input image | — | — | | | | Conv | | 1 | | | | Conv | | 2 | | | | Conv | | 1 | | | | Conv | | 2 | | | | Conv | | 1 | | | | Route 4 | — | — | — | | | Conv | | 1 | | | | Conv | | 1 | | | | Conv | | 1 | | | | Conv | | 1 | | | | Conv | | 1 | | | | Route 5 7 9 11 | — | — | — | | | Conv | | 1 | | | | MaxPool | — | — | | | | Conv | | 1 | | | | Route 13 | — | — | — | | | Conv | | 1 | | | | Conv | | 2 | | | | Route 15 18 | — | — | — | | | Conv | | 1 | | | | Route 19 | — | — | — | | | Conv | | 1 | | | | Conv | | 1 | | | | Conv | | 1 | | | | Conv | | 1 | | | | Conv | | 1 | | | | Route 20 22 24 26 | — | — | — | | | Conv | | 1 | | | | Conv | | 1 | | | | Route 28 | — | — | — | | | MaxPool | — | — | | | | Conv | | 1 | | | | Route 28 | — | — | — | | | Conv | | 1 | | | | Conv | | 2 | | | | Route 32 35 | — | — | — | | | Conv | | 1 | | | | Route 36 | — | — | — | | | Conv | | 1 | | | | Conv | | 1 | | | | Conv | | 1 | | | | Conv | | 1 | | | | Conv | | 1 | | | | Route 37 39 41 43 | — | — | — | | | Conv | | 1 | | | | Conv | | 1 | | | | Route 45 | — | — | — | | | MaxPool | — | — | | | | Conv | | 1 | | | | Route 45 | — | — | — | | | Conv | | 1 | | | | Conv | | 2 | | | | Route 49 52 | — | — | — | | | Conv | | 1 | | | | Route 53 | — | — | — | | | Conv | | 1 | | | | Conv | | 1 | | | | Conv | | 1 | | | | Conv | | 1 | | | | Conv | | 1 | | | | Route 54 56 58 60 | — | — | — | | | Conv | | 1 | | | | Conv | | 1 | | | | Route 62 | — | — | — | | | Conv | | 1 | | | | Conv | | 1 | | | | Conv | | 1 | | | | MaxPool | — | — | | | | Route 67 | — | — | — | | | MaxPool | — | — | | | | Route 67 | — | — | — | | | MaxPool | — | — | | | | Route 67 68 70 72 | — | — | — | | | Conv | | 1 | | | | Conv | | 1 | | | | Route 63 75 | — | — | — | | | Conv | | 1 | | | | Conv | | 1 | | | | Upsample | — | — | | | | Route 46 79 | — | — | — | | | Conv | | 1 | | | | Route 80 | — | — | — | | | Conv | | 1 | | | | Conv | | 1 | | | | Conv | | 1 | | | | Conv | | 1 | | | | Conv | | 1 | | | | Route 81 83 85 87 | — | — | — | | | Conv | | 1 | | | | Conv | | 1 | | | | Upsample | — | — | | | | Route 29 91 | — | — | — | | | Conv | | 1 | | | | Route 92 | — | — | — | | | Conv | | 1 | | | | Conv | | 1 | | | | Conv | | 1 | | | | Conv | | 1 | | | | Conv | | 1 | | | | Route 93 95 97 99 | — | — | — | | | Conv | | 1 | | | | MaxPool | — | — | | | | Conv | | 1 | | | | Route 101 | — | — | — | | | Conv | | 1 | | | | Conv | | 2 | | | | Route 103 106 | — | — | — | | | Route 89 107 | — | — | — | | | Conv | | 1 | | | | Route 108 | — | — | — | | | Conv | | 1 | | | | Conv | | 1 | | | | Conv | | 1 | | | | Conv | | 1 | | | | Conv | | 1 | | | | Route 109 111 113 115 | — | — | — | | | Conv | | 1 | | | | MaxPool | — | — | | | | Conv | | 1 | | | | Route 117 | — | — | — | | | Conv | | 1 | | | | Conv | | 2 | | | | Route 119 122 | — | — | — | | | Route 77 123 | — | — | — | | | Conv | | 1 | | | | Route 124 | — | — | | | | Conv | | 1 | | | | Conv | | 1 | | | | Conv | | 1 | | | | Conv | | 1 | | | | Conv | | 1 | | | | Route 125 127 129 131 | — | — | — | | | Conv | | 1 | | | | Conv | | 1 | | | | Conv | | 1 | | | | Conv | | 1 | | | | Conv | | 1 | | | | Route 133 | — | — | — | | | AvgPool | | | | | | Conv | | 1 | | | | Conv | | 1 | | | | Route 133 | — | — | — | | | MaxPool | — | — | | | | Conv | | 1 | | | | Conv | | 1 | | | | Add 141 145 | — | — | — | | | Mul 133 146 | — | — | — | | | Add 137 147 | — | — | — | | | Mul 133 148 | — | — | — | | | MaxPool | — | — | | | | Route 149 | — | — | — | | | AvgPool | — | — | | | | Route 150 152 | — | — | — | | | Conv | | 1 | | | | Mul 149 154 | — | — | — | | | Add 133 155 | — | — | — | | | Route 133 | — | — | — | | | Conv | | 1 | | | | Conv | | 1 | | | | Conv | | 1 | | | | Conv | | 1 | | | | Route 133 | — | — | — | | | AvgPool | — | — | | | | Conv | | 1 | | | | Conv | | 1 | | | | Route 133 | — | — | — | | | MaxPool | — | — | | | | Conv | | 1 | | | | Conv | | 1 | | | | Add 165 169 | — | — | — | | | Mul 133 170 | — | — | — | | | Add 161 171 | — | — | — | | | Mul 133 172 | — | — | — | | | MaxPool | — | — | | | | Route 173 | — | — | | | | AvgPool | — | — | | | | Route 174 176 | — | — | — | | | Conv | | 1 | | | | Mul 173 178 | — | — | — | | | Add 133 179 | — | — | — | | | Add 156 180 | — | — | — | | | RepConv | | 1 | | | | YOLO | | | | | | Route 117 | — | — | — | | | Conv | | 1 | | | | Conv | | 1 | | | | Conv | | 1 | | | | Conv | | 1 | | | | Route 117 | — | — | — | | | AvgPool | — | — | | | | Conv | | 1 | | | | Conv | | 1 | | | | Route 117 | — | — | — | | | MaxPool | — | — | | | | Conv | | 1 | | | | Conv | | 1 | | | | Add 192 196 | — | — | — | | | Mul 117 197 | — | — | — | | | Add 188 198 | — | — | — | | | Mul 117 199 | — | — | — | | | MaxPool | — | — | | | | Route 200 | — | — | — | | | AvgPool | — | — | | | | Route 201 203 | — | — | — | | | Conv | | 1 | | | | Mul 200 205 | — | — | — | | | Add 117 206 | — | — | — | | | Route 117 | — | — | — | | | Conv | | 1 | | | | Conv | | 1 | | | | Conv | | 1 | | | | Conv | | 1 | | | | Route 117 | — | — | — | | | AvgPool | — | — | | | | Conv | | 1 | | | | Conv | | 1 | | | | Route 117 | — | — | — | | | MaxPool | — | — | | | | Conv | | 1 | | | | Conv | | 1 | | | | Add 216 220 | — | — | — | | | Mul 117 221 | — | — | — | | | Add 212 222 | — | — | — | | | Mul 117 223 | — | — | — | | | MaxPool | — | — | | | | Route 224 | — | — | — | | | AvgPool | — | — | | | | Route 225 227 | — | — | — | | | Conv | | 1 | | | | Mul 224 229 | — | — | — | | | Add 117 230 | — | — | — | | | Add 207 231 | — | — | — | | | RepConv | | 1 | | | | YOLO | | | | | | Route 101 | — | — | — | | | Conv | | 1 | | | | Conv | | 1 | | | | Conv | | 1 | | | | Conv | | 1 | | | | Route 101 | — | — | — | | | AvgPool | — | — | | | | Conv | | 1 | | | | Conv | | 1 | | | | Route 101 | — | — | — | | | MaxPool | — | — | | | | Conv | | 1 | | | | Conv | | 1 | | | | Add 243 247 | — | — | — | | | Mul 101 248 | — | — | — | | | Add 239 249 | — | — | — | | | Mul 101 250 | — | — | — | | | MaxPool | — | — | | | | Route 251 | — | — | — | | | AvgPool | — | — | | | | Route 252 254 | — | — | — | | | Conv | | 1 | | | | Mul 251 256 | — | — | — | | | Add 101 257 | — | — | — | | | Route 101 | — | — | — | | | Conv | | 1 | | | | Conv | | 1 | | | | Conv | | 1 | | | | Conv | | 1 | | | | Route 101 | — | — | — | | | AvgPool | — | — | | | | Conv | | 1 | | | | Conv | | 1 | | | | Route 101 | — | — | — | | | MaxPool | — | — | | | | Conv | | 1 | | | | Conv | | 1 | | | | Add 267 271 | — | — | — | | | Mul 101 272 | — | — | — | | | Add 263 273 | — | — | — | | | Mul 101 274 | — | — | — | | | MaxPool | — | — | | | | Route 275 | — | — | — | | | AvgPool | — | — | | | | Route 276 278 | — | — | — | | | Conv | | 1 | | | | Mul 275 279 | — | — | — | | | Add 101 280 | — | — | — | | | Add 258 281 | — | — | — | | | RepConv | | 1 | | | | YOLO | | | | |
|
|
