这两天玩了一下stable diffusion,是真的好玩!
然后我在想遥感有没有相关的生成模型,找了一下,还真找到了(https://github.com/xiaoyuan1996/Stable-Diffusion-for-Remote-Sensing-Image-Generation/tree/main)。
该模型的作者是空天院的Zhiqiang yuan,他提供了相关的训练代码、预测代码、已经训练好的模型等。
[外链图片转存中…(img-lOFMZR6f-1705867377350)]
因为我也不想装代码环境,就只下载了模型加载到stable diffusion中看看生成的遥感影像的效果。
感兴趣的读者可以自己跑一下他的代码。
[外链图片转存中…(img-YYwr4ecU-1705867377351)]
[外链图片转存中…(img-fgKLzarg-1705867377351)]
[外链图片转存中…(img-cj2Jvx8A-1705867377351)]
[外链图片转存中…(img-zYIHB4Bu-1705867377352)]
我只是一个初次尝试stable diffusion的使用者,有被这个效果惊艳到。有兴趣的话,可以看看这个模型作者的相关论文。
代码和数据:https://github.com/xiaoyuan1996/Stable-Diffusion-for-Remote-Sensing-Image-Generation/tree/main
相关论文:Yuan Z, Hao C, Zhou R, et al. Efficient and Controllable Remote Sensing Fake Sample Generation Based on Diffusion Model[J]. IEEE Transactions on Geoscience and Remote Sensing, 2023.