Nowadays in UAV and remote sensing image processing area, rotated object detection has been attached more and more attentions. However, there are few studies of this based on Transformer, which is much stronger than CNN for feature extraction. Moreover, the well-developed IoU based loss function in horizontal object detection area is not well fits with rotated objects. In this paper, we argue that Transformer of pyramid structure called Swin-Transformer is an effective alternative of CNN. Specifically, the finetuned Swin-Transformer with Relatively Position Encoding (RPE) performed much better than other backbones generally used. Moreover, the new kind of IoU loss called Gaussian Estimate Loss (GE Loss) that use gaussian kernel to model object is applied in our model. It can increase the precious of the model. This loss is in contract to other Gaussian modeling loss function for adding direction vector that can solve the difficulty of testing objects close to square. Experiments on DOTA dataset achieved 84.99% map, which indicates that and our experiment shows that these improvements of our model are effective.
IOS Press, Inc.
6751 Tepper Drive
Clifton, VA 20124
Tel.: +1 703 830 6300
Fax: +1 703 830 2300 firstname.lastname@example.org
(Corporate matters and books only) IOS Press c/o Accucoms US, Inc.
For North America Sales and Customer Service
West Point Commons
Lansdale PA 19446
Tel.: +1 866 855 8967
Fax: +1 215 660 5042 email@example.com