Research Article
OAB-YOLOv5: One-Anchor-Based YOLOv5 for Rotated Object Detection in Remote Sensing Images
Table 2
Performance comparisons on DOTA-v1.5 test (OBB task) (%).
| OBB results | Methods | Params | PL | BD | BR | GTF | SV | LV | SH | TC | BC | ST | SBF | RA | HA | SP | HC | CC | mAP |
| DCL [36] | 33.4 M | 79.09 | 72.7 | 37.85 | 62.67 | 46.91 | 50.65 | 73.21 | 89.41 | 72.53 | 59.62 | 51.99 | 68.81 | 52.36 | 65.56 | 56.49 | 10.23 | 59.40 | RSDet [23] | 33.2 M | 79.26 | 79.67 | 41.61 | 67.03 | 48.42 | 53.41 | 73.55 | 89.30 | 75.08 | 63.41 | 50.65 | 68.50 | 61.96 | 64.27 | 55.08 | 11.44 | 64.40 | GWD [37] | 33.1 M | 79.39 | 74.46 | 41.95 | 60.15 | 49.97 | 60.03 | 76.17 | 90.31 | 71.75 | 58.48 | 48.04 | 67.8 | 55.8 | 65.79 | 51.35 | 9.99 | 60.00 | KLD [38] | 41.8 M | 80.31 | 74.33 | 47.87 | 60.14 | 65.98 | 73.11 | 87.1 | 89.55 | 75.38 | 81.80 | 49.98 | 71.8 | 64.34 | 71.75 | 55.97 | 0.00 | 65.60 | R2-CNN [39] | 41.8 M | 80.38 | 78.7 | 47.95 | 62.55 | 65.6 | 71.33 | 86.34 | 89.74 | 76.29 | 76.23 | 49.73 | 67.55 | 63.47 | 73.14 | 58.44 | 15.66 | 66.40 | Rotate-YOLOv5 [29] | 41.8 M | 87.82 | 78.13 | 45.60 | 58.73 | 66.70 | 76.21 | 88.69 | 90.69 | 79.59 | 78.05 | 41.70 | 70.40 | 65.61 | 75.47 | 54.53 | 10.86 | 66.80 | RetinaNet [14] | — | 71.43 | 77.64 | 42.12 | 64.65 | 44.53 | 56.79 | 73.31 | 90.84 | 76.02 | 59.96 | 46.95 | 69.24 | 59.65 | 64.52 | 48.06 | 0.83 | 59.16 | MR [40] | — | 76.84 | 73.51 | 49.9 | 57.8 | 51.31 | 71.34 | 79.75 | 90.46 | 74.21 | 66.07 | 46.21 | 70.61 | 63.07 | 64.46 | 57.81 | 9.42 | 62.67 | CMR [41] | — | 77.77 | 74.62 | 51.09 | 63.44 | 51.64 | 72.9 | 79.99 | 90.35 | 74.9 | 67.58 | 49.54 | 72.85 | 64.19 | 64.88 | 55.87 | 3.02 | 63.41 | FR OBB [21] | — | 71.89 | 74.47 | 44.45 | 59.87 | 51.28 | 68.98 | 79.37 | 90.78 | 77.38 | 67.5 | 47.75 | 69.72 | 61.22 | 65.28 | 60.47 | 1.54 | 62.00 | FR H-OBB [21] | — | 71.57 | 74.71 | 46.39 | 63.4 | 51.54 | 70.11 | 79.09 | 90.63 | 76.81 | 67.4 | 48.66 | 70.9 | 63.1 | 65.67 | 56.66 | 4.55 | 62.57 | FR OBB + RT [23] | — | 71.92 | 76.07 | 51.87 | 69.24 | 52.05 | 75.18 | 80.72 | 90.53 | 78.58 | 68.26 | 49.18 | 71.74 | 67.51 | 65.53 | 62.16 | 9.99 | 65.03 | OAB-YOLOv5 (ours) | 20.3 M | 80.67 | 77.2 | 46.37 | 60.13 | 66.08 | 75.80 | 87.85 | 90.64 | 78.12 | 79.54 | 46.28 | 70.58 | 67.70 | 75.42 | 53.63 | 28.15 | 68.02 |
|
|
The short names for categories are defined as follows: PL: plane; BD: baseball diamond; BR: bridge; GTF: ground field track; SV: small vehicle; LV: large vehicle; SH: ship; TC: tennis court; BC: basketball court; ST: storage tank; SBF: soccerball field; RA: roundabout; HA: harbor; SP: swimming pool; HC: helicopter; CC: container crane.
|