【文件属性】:
文件名称:mxnet_tetris:基于mxnet训练rnn网络, 玩俄罗斯方块
文件大小:146KB
文件格式:ZIP
更新时间:2021-05-12 07:31:28
Python
训练rnn网络,自己玩俄罗斯方块
通过训练一个GRU网络,学习“一个新的形状出现到形状不能动为止”的用户键盘操作序列,让模型自己玩俄罗斯方块游戏。
基本思路
俄罗斯方块本质是一个二位数组,这个可以认为是一个单通道的“image”,用户的键盘输入则是当前 image 下的“label”。
因此可以有两种思路:
图像分类:输入游戏的当前数组作为一副图像,输出的分类标签就是当前的键盘操作(左,右,下,旋转...)
rnn序列预测:从形状出现到结束,对应这一个用户键盘的操作序列,照理说,这种序列比静态的图像分类有更强的鲁棒性;
尝试一下?
生成样本:
生成训练样本, 将在 rnn_train 子目录中生成大量npz文件, 每个npz中存储了图像+按键序列.
python game.py train
生成测试样本
python game.py val
训练
python train_rnn.py
一
【文件预览】:
mxnet_tetris-master
----db.py(2KB)
----train.py(5KB)
----show.py(2KB)
----main.py(247B)
----shape.py(7KB)
----README.md(1KB)
----background.jpg(126KB)
----inference_rnn.py(7KB)
----game.py(24KB)
----fix_label.py(836B)
----show_sample.py(769B)
----train_rnn.py(10KB)