图像(数学)
计算机科学
扩散
计算机图形学(图像)
人工智能
计算机视觉
物理
热力学
作者
Xin Zhang,Yang Li,Feng Long,Hangzhi Jiang,Yanhua Wang,Liang Zhang,Lirong Zheng,Zegang Ding
出处
期刊:Isprs Journal of Photogrammetry and Remote Sensing
日期:2024-01-01
卷期号:207: 203-217
被引量:1
标识
DOI:10.1016/j.isprsjprs.2023.12.002
摘要
We present Ship-Go, an instance-to-image diffusion model, to increase the scale and diversity of the SAR detection datasets. Ship-Go is developed as a multi-conditions denoising diffusion probabilistic model (DDPM), i.e., it takes the proposed visual instances and environment prompt condition variables as constraints to generate backscatter intensity information for each resolution cell, resulting in inpainting a SAR image and generating the corresponding instance-level annotations, which can be directly employed to train detection models. We demonstrate, for the first time, the ability to place the ship objects at any angle, size, and arrangement in the generated background of multiple specified environment types. Importantly, two image generation scenarios are designed to increase the diversity of objects and backgrounds for the original dataset (i.e., in-domain augmentation and out-of-domain augmentation). The experiment verifies that the generated detection datasets boost the performance of multiple classical deep detectors in the different cases of insufficient samples. Qualitatively, we find that Ship-Go improves the diversity of the existing dataset, and has sample distribution transferability among multiple datasets. Code and models are available at: https://github.com/XinZhangRadar/Ship-Go.
科研通智能强力驱动
Strongly Powered by AbleSci AI