PistonNet: Object Separating From Background by Attention for Weakly Supervised Ship Detection

Object detection under weakly supervised learning is a challenging issue. In remote sensing ship detection task, the weakly supervised learning method requires that the training set only has image-level class annotations. In the absence of location information, it is difficult to locate ships and extract features. Moreover, when the detector extracts more than one candidate region, the image-level annotation makes it difficult to determine whether the candidate regions are multiple ships or mixed with the detected background. To address these issues, this article analyzes the interaction between class information and location information, and proposes a weakly supervised detection method, PistonNet, based on Data-efficient image Transformers (DeiT). PistonNet proposes an artificial point, which is inserted into the feature map. Artificial point can suppress the background and enhance the object by interfering in the weight distribution of object and background in Self-Attention calculation. PistonNet also proposes joint confidence probability to improve detection accuracy. Experiments on the GF1-LRSD and NWPU VHR-10 dataset show that the proposed method boosts detection performance effectively. The main improvements of PistonNet as well as the contributions of this article are threefold. Firstly, PistonNet is a specific weakly supervised object detection (WSOD) method for single-class detection, which provides an innovative approach to classifying target and background. Secondly, PistonNet reaches the level of advanced supervised detectors on detection accuracy with fewer parameters. Finally, the objects’ locations detected by PistonNet are obtained by segmenting regions on the heatmap. PistonNet’s background suppression ability makes it free from dependence on segmentation threshold.