|
Object detection is a critical research topic in computer vision, and its research results have been widely used in recent years. As a subtask of object detection, Ship detection has important research significance. Most ship detection research is based on SAR (Synthetic Aperture Radar) images in the existing research. However, the imaging method of SAR image is different from that of optical image, and it is impossible to transfer the research results of SAR image to optical image. Compared with SAR images, optical images have more image feature information, which can assist the algorithm to better learn ship features. In addition, the research of optical image ship detection has more important commercial value. Companies only need to equip a simple optical camera to complete the ship detection, without the need for a valuable device like radar, which is more reusable. This paper conducts optical image ship detection experiments, applies the YOLO v3 algorithm to the ship detection task, introduces the attention mechanism to transform the residual block in the DarkNet-53 network, and achieves more excellent performance. At the same time,this paper optimizes the recently proposed CIoU loss function which is better than the ln-norm loss function and presents the AIoU loss function on this basis. By increasing the area penalty term, the AIoU loss improves performance while converging speed compared with CIoU loss. It is applied to ship detection in the optical image and compared with GIoU, DIoU, and CIoU loss functions, and it has achieved better results than them in the bounding box regression of ship detection. |
|
Keywords:ship detection, attention mechanism, residual block, loss function, boundingbox regression |
|