简介
这两天玩了一下stable diffusion,是真的好玩!
然后我在想遥感有没有相关的生成模型,找了一下,还真找到了(https://github.com/xiaoyuan1996/Stable-Diffusion-for-Remote-Sensing-Image-Generation/tree/main)。
该模型的作者是空天院的Zhiqiang yuan,他提供了相关的训练代码、预测代码、已经训练好的模型等。
[外链图片转存中…(img-lOFMZR6f-1705867377350)]
生成效果
因为我也不想装代码环境,就只下载了模型加载到stable diffusion中看看生成的遥感影像的效果。
感兴趣的读者可以自己跑一下他的代码。
(1)指令:Some boats drived in the sea
(2)指令:There is a church in a dark green forest with two table tennis courts next to it
(3)指令:Many houses
(4)指令:There are many farmlands and some trees
[外链图片转存中…(img-YYwr4ecU-1705867377351)]
(5)指令:There is a square surrounded by deserts
[外链图片转存中…(img-fgKLzarg-1705867377351)]
(6)指令:A small amount of arable land in a snowy mountain and a lake
[外链图片转存中…(img-cj2Jvx8A-1705867377351)]
(7)指令:A long river
[外链图片转存中…(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.