通过Stable Diffusion生成虚假的遥感影像

简介

这两天玩了一下stable diffusion,是真的好玩!

然后我在想遥感有没有相关的生成模型,找了一下,还真找到了(https://github.com/xiaoyuan1996/Stable-Diffusion-for-Remote-Sensing-Image-Generation/tree/main)。

通过Stable Diffusion生成虚假的遥感影像_第1张图片

该模型的作者是空天院的Zhiqiang yuan,他提供了相关的训练代码、预测代码、已经训练好的模型等。

[外链图片转存中…(img-lOFMZR6f-1705867377350)]

生成效果

因为我也不想装代码环境,就只下载了模型加载到stable diffusion中看看生成的遥感影像的效果。

感兴趣的读者可以自己跑一下他的代码。

(1)指令:Some boats drived in the sea

通过Stable Diffusion生成虚假的遥感影像_第2张图片

(2)指令:There is a church in a dark green forest with two table tennis courts next to it

(3)指令:Many houses

通过Stable Diffusion生成虚假的遥感影像_第3张图片

(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.

你可能感兴趣的:(stable,diffusion)