护踝厂家
免费服务热线

Free service

hotline

010-00000000
护踝厂家
热门搜索:
行业资讯
当前位置:首页 > 行业资讯

如何用TensorFlow生成令人惊艳的分形图案-【新闻】

发布时间:2021-04-05 22:57:07 阅读: 来源:护踝厂家

雷锋网 AI科技评论按:本文作者何之源,原文载于知乎专栏AI Insight, AI科技评论获其授权发布。

今天来介绍一个小项目:在TensorFlow中生成分形图案。分形本身只是一个数学概念,与机器学习并无太大关系,但是通过分形的生成,我们可以了解怎么在TensorFlow中进行数学计算,以及如何进行基本的流程控制,是学习TensorFlow的一个非常好的练手项目。

在开始之前,需要说明的是,TensorFlow官方也提供了一个生成分形图案的教程(地址: www.tensorflow.org/tutorials/mandelbrot),然而官方教程中生成的图像实在是太丑了,而且只能生成一种图案,我对官方的代码做了一些改进,并且加入了多种类型的分形,此外,不仅可以生成图像,还可以制作gif动画,代码已经放到了Github上:https://github.com/hzy46/tensorflow-fractal-playground,主要的程序只有50行,欢迎大家参考。

Mandelbrot集合

Mandelbrot集合是分形中最经典的一个例子。考虑迭代公式 (z和c都是复数)。当 为0时,得到的值可以组成一个数列,依次为 。当该数列发散到无穷时,对应的点就属于Mandelbrot集合。

如 时,显然数列永远是0,并不发散,因此0不属于Mandelbrot集合。

又如 时,对应的数列为 ,数字越来越庞大,因此3i就属于Mandelbrot集合。

在二维平面上,将所有不属于Mandelbrot集合的点标记为黑色,将所有属于Mandelbrot集合的点按照其发散速度赋予不同的颜色,就可以得到Mandelbrot的经典图像:

<img src="https://static.leiphone.com/uploads/new/article/pic/201709/fdc6823436bde5a6ea991c1931747893.png" data-rawwidth="1000" data-rawheight="686" class="origin_image zh-lightbox-thumb" width="1000" data-original="https://pic1.zhimg.com/v2-ffca32a868fc484837cd7bd17de6102c_r.png" _src="https://static.leiphone.com/uploads/new/article/pic/201709/fdc6823436bde5a6ea991c1931747893.png"/>

上面这张图完全是使用TensorFlow进行计算的,类似的图大家应该在网上也见过好多了,在TensorFlow中,我们定义下面的计算步骤:

xs = tf.constant(Z.astype(np.complex64))

zs = tf.Variable(xs)

ns = tf.Variable(tf.zeros_like(xs, tf.float32)) with tf.Session():

tf.global_variables_initializer().run()

zs_ = tf.where(tf.abs(zs) < R, zs**2 + xs, zs)

not_diverged = tf.abs(zs_) < R

step = tf.group(

zs.assign(zs_),

ns.assign_add(tf.cast(not_diverged, tf.float32))

for i in range(ITER_NUM): step.run()

final_step = ns.eval()

final_z = zs_.eval()

zs就对应我们之前迭代公式的z,而xs就对应迭代公式中的c。为了方便起见,只要计算时数值的绝对值大于一个事先指定的值R,就认为其发散。每次计算使用tf.where只对还未发散的值进行计算。结合ns和zs_就可以计算颜色,得到经典的Mandelbrot图像。

Julia集合

Julia集合和Mandelbrot集合差不多,但这次我们固定c,转而计算发散的z的值。即c是固定的常数(可以任取),数列变成 。如果该数列发散,对应的z就属于Julia集合。对此,我们只要在原来的程序中修改两行内容,就可以生成Julia集合:

xs = tf.constant(np.full(shape=Z.shape, fill_value=c, dtype=Z.dtype))

zs = tf.Variable(Z)

我们在fill_value=c处指定了Julia集合中的c值,只要使用不同的c值,就可以生成完全不同的Julia集合!

默认: :

&amp;amp;lt;img src=&amp;amp;quot;https://static.leiphone.com/uploads/new/article/pic/201709/189e1749a0f7b44bd35b64923c1bc6c8.png&amp;amp;quot; data-rawwidth=&amp;amp;quot;1200&amp;amp;quot; data-rawheight=&amp;amp;quot;695&amp;amp;quot; class=&amp;amp;quot;origin_image zh-lightbox-thumb&amp;amp;quot; width=&amp;amp;quot;1200&amp;amp;quot; data-original=&amp;amp;quot;https://pic3.zhimg.com/v2-fa34eacb1ad6c4eebb7c3d59398c5e72_r.png&amp;amp;quot; _src=&amp;amp;quot;https://static.leiphone.com/uploads/new/article/pic/201709/189e1749a0f7b44bd35b64923c1bc6c8.png&amp;amp;quot;/&amp;amp;gt;

将c值变为 ,并调整颜色(调整方法参考Github页面的说明):

&amp;amp;lt;img src=&amp;amp;quot;https://static.leiphone.com/uploads/new/article/pic/201709/9f9146f45067fa89aa38bf6496273a70.png&amp;amp;quot; data-rawwidth=&amp;amp;quot;1200&amp;amp;quot; data-rawheight=&amp;amp;quot;1040&amp;amp;quot; class=&amp;amp;quot;origin_image zh-lightbox-thumb&amp;amp;quot; width=&amp;amp;quot;1200&amp;amp;quot; data-original=&amp;amp;quot;https://pic3.zhimg.com/v2-1c88aff1df12798b6a2b552079b7098e_r.png&amp;amp;quot; _src=&amp;amp;quot;https://static.leiphone.com/uploads/new/article/pic/201709/9f9146f45067fa89aa38bf6496273a70.png&amp;amp;quot;/&amp;amp;gt;

选用 ,图案又变得完全不同:

&amp;amp;lt;img src=&amp;amp;quot;https://static.leiphone.com/uploads/new/article/pic/201709/3542c2f2c23a08d0b3b7452c51bc0d99.png&amp;amp;quot; data-rawwidth=&amp;amp;quot;1200&amp;amp;quot; data-rawheight=&amp;amp;quot;1040&amp;amp;quot; class=&amp;amp;quot;origin_image zh-lightbox-thumb&amp;amp;quot; width=&amp;amp;quot;1200&amp;amp;quot; data-original=&amp;amp;quot;https://pic1.zhimg.com/v2-3647ec260c639c35e37972dfad30fbd8_r.png&amp;amp;quot; _src=&amp;amp;quot;https://static.leiphone.com/uploads/new/article/pic/201709/3542c2f2c23a08d0b3b7452c51bc0d99.png&amp;amp;quot;/&amp;amp;gt;

生成Julia集合的动画

在Julia集合中,每次都对c的值进行微小的改变,并将依次生成图片制作为gif,就可以生成如下所示的动画,对应的代码为julia_gif.py:

这里由于上传gif有大小限制的关系,只展示了一个小尺寸的动画图像。程序中提供了一个width参数,可以修改它以生成更大尺寸,质量更高的动画图像。

探索Mandelbrot集合

(注意:下面的图片可能对密集恐惧症患者不太友好。。。因此慎重翻页。。)

在前面生成的Mandelbrot集合中,我们可以将图像放大,选取某些区域进行生成,就可以得到格式各样造型迥异的分形图案,对应的程序为mandelbrot_area.py。

在Mandelbrot集合中,有很多地方图案比较奇特,如下图中的9个位置。

&amp;amp;lt;img src=&amp;amp;quot;https://static.leiphone.com/uploads/new/article/pic/201709/765b4cce7bfca4098c1caba17cb10d7c.gif&amp;amp;quot; data-rawwidth=&amp;amp;quot;200&amp;amp;quot; data-rawheight=&amp;amp;quot;150&amp;amp;quot; data-thumbnail=&amp;amp;quot;https://static.leiphone.com/uploads/new/article/pic/201709/91ad333ebfbbd704323e532e3c0a543a.jpg&amp;amp;quot; class=&amp;amp;quot;content_image&amp;amp;quot; width=&amp;amp;quot;200&amp;amp;quot; _src=&amp;amp;quot;https://static.leiphone.com/uploads/new/article/pic/201709/765b4cce7bfca4098c1caba17cb10d7c.gif&amp;amp;quot;/&amp;amp;gt;

其中编号为2的地方被称为“Elephant Valley”,因为此处的图案与大象很像,直接运行mandelbrot_area.py就可以得到该区域的图像:

&amp;amp;lt;img src=&amp;amp;quot;https://static.leiphone.com/uploads/new/article/pic/201709/d396fe230bbb0b28b8476219e51ca90c.png&amp;amp;quot; data-rawwidth=&amp;amp;quot;1000&amp;amp;quot; data-rawheight=&amp;amp;quot;800&amp;amp;quot; class=&amp;amp;quot;origin_image zh-lightbox-thumb&amp;amp;quot; width=&amp;amp;quot;1000&amp;amp;quot; data-original=&amp;amp;quot;https://pic4.zhimg.com/v2-936fa16a9362e7f350e15a2c695e2713_r.png&amp;amp;quot; _src=&amp;amp;quot;https://static.leiphone.com/uploads/new/article/pic/201709/d396fe230bbb0b28b8476219e51ca90c.png&amp;amp;quot;/&amp;amp;gt;

编号为3的地方被称为“Triple Spiral Valley”(三重螺旋),在mandelbrot_area.py修改一下坐标位置为(ratio调整的是颜色):

start_x = -0.090 # x range

end_x = -0.086

start_y = 0.654 # y range

end_y = 0.657

width = 1000

ratio1, ratio2, ratio3 = 0.2, 0.6, 0.6

就可以得到该处的图案:

&amp;amp;lt;img src=&amp;amp;quot;https://static.leiphone.com/uploads/new/article/pic/201709/2633f166ac043e12dd0b9c1e3f2c312d.png&amp;amp;quot; data-rawwidth=&amp;amp;quot;1001&amp;amp;quot; data-rawheight=&amp;amp;quot;751&amp;amp;quot; class=&amp;amp;quot;origin_image zh-lightbox-thumb&amp;amp;quot; width=&amp;amp;quot;1001&amp;amp;quot; data-original=&amp;amp;quot;https://pic2.zhimg.com/v2-e3303f438522410d5db4c97b0b6ebfa9_r.png&amp;amp;quot; _src=&amp;amp;quot;https://static.leiphone.com/uploads/new/article/pic/201709/2633f166ac043e12dd0b9c1e3f2c312d.png&amp;amp;quot;/&amp;amp;gt;

最后编号为1的地方被称为“Seahorse Valley”(海马山谷),对应的坐标为:

start_x = -0.750 # x range

end_x = -0.747

start_y = 0.099 # y range

end_y = 0.102

width = 1000

ratio1, ratio2, ratio3 = 0.1, 0.1, 0.3

图像如下,确实和海马有一点神似:

&amp;amp;lt;img src=&amp;amp;quot;https://static.leiphone.com/uploads/new/article/pic/201709/084cc607c6e95f73777564bd4b0021ba.png&amp;amp;quot; data-rawwidth=&amp;amp;quot;1000&amp;amp;quot; data-rawheight=&amp;amp;quot;1000&amp;amp;quot; class=&amp;amp;quot;origin_image zh-lightbox-thumb&amp;amp;quot; width=&amp;amp;quot;1000&amp;amp;quot; data-original=&amp;amp;quot;https://pic1.zhimg.com/v2-ddec205c425dac242aba103c5e2e4bc0_r.png&amp;amp;quot; _src=&amp;amp;quot;https://static.leiphone.com/uploads/new/article/pic/201709/084cc607c6e95f73777564bd4b0021ba.png&amp;amp;quot;/&amp;amp;gt;

生成更多的图案

项目提供了两个jupyter notebook:Mandelbrot.ipynb和Julia.ipynb可以对Mandelbrot集合、Julia集合做更方便的探索。其中,Mandelbrot集的更多坐标位置可以参考Quick Guide to the Mandelbrot Set(http://www.nahee.com/Derbyshire/manguide.html),Julia集中更多有趣的c值可以参考Julia set - Wikipedia(https://en.wikipedia.org/wiki/Julia_set#Quadratic_polynomials)。网上类似的资源还有很多。

最后再安利一下项目地址:https://github.com/hzy46/tensorflow-fractal-playground。如果代码有什么问题可以直接发在评论里或者在Github上提出issue:)

雷锋网版权文章,未经授权禁止转载。详情见转载须知。

秦时明月

猎魂觉醒官服下载

寒灵之剑手游

桃源仙境

相关阅读