我发现使用Tensorflow https://github.com/asrivat1/DeepLearningVideoGames进行深度学习的实验性使用。作者训练CNN玩Pong游戏。除了可视化以说明CNN层中的Q值之外,对我来说似乎都很简单。这是YouTube视频https://www.youtube.com/watch?v=W9jGIzkVCsM。任何人都可以解释如何绘制图表(热图看)?
THX。
答案 0 :(得分:2)
我深入研究代码并从之前的提交中找到了this file,但它在主版本中不再存在(奇怪)。
在里面你会找到可视化的代码,重要的lines是:
self.l1.imshow(np.reshape(np.rollaxis(c1, 2, 1),(20,20*32)),aspect = 6)
self.l2.imshow(np.reshape(np.rollaxis(c2, 2, 1),(5,5*64)),aspect = 12)
self.l3.imshow(np.reshape(np.rollaxis(c3, 2, 1),(3,3*64)),aspect = 12)
这里他们采用大小(20,20,32)的激活图并绘制所有激活图。它们重塑为(20, 20*32)
以绘制所有要素图(总共32个)并排绘制。为了使其适合屏幕,它们使用宽高比6,水平压缩图像。
总结一下,他们并排绘制所有要素图,并将其压缩以适应屏幕。 我建议你不要改变宽高比,而是每次激活时使用小块(总共32块),并以8x4布局排列块。