标签：处理 gym obs 灰度 atari 打卡 预处理 84

最近参加完百度强化学习7日打卡营的学习班，初步了解了一些强化学习的算法。学习完后，试着开始玩玩atari的游戏，老师给了一个demo，https://github.com/PaddlePaddle/PARL/tree/develop/examples/DQN_variant，model，algorithm，agent等就不多说了，这些老师课堂已经讲的很清楚了。

这个demo，需要atari游戏的rom，我是上天入地的找，找了半天都没合适的，但是gym的gym.make的游戏倒是非常多，一开始我直接gym.make，但是程序报错，alueError: all the input arrays must have same number of dimensions，看到input arrays 我第一反应应该是obs出了问题，果不其然，demo里对rom进行了处理，出来的obs的shape是（84,84），而我这里是（84,84,3），于是便开始对图形进行灰度处理，我在obs的地方进行了灰度处理，程序ok，但是运行了test的时候又报错了，讲道理，应该是没有问题的，因为我对所有的obs都进行了处理，但是无济于事，于是还是选择从env = gym.make(game_name)处进行处理，demo里atari_wrapper提供了对env的一些处理，例如图像缩放，帧数重复的处理等等，维度没有进行灰度处理，好说写了一个灰度处理的方法：

def rgb2gray(image):
return np.dot(image[...,:3], [0.299, 0.587, 0.114])

然后通过atari_wrapper的mapstate方法，直接对其进行处理：

env = MapState(env, lambda im: rgb2gray(im))

，程序OK，经过改造，程序可以直接使用gym的游戏，gym调用游戏非常方便，而且很多，直接make一个就可以了。

标签：处理,gym,obs,灰度,atari,打卡,预处理,84
来源： https://www.cnblogs.com/martinho/p/13196313.html

本站声明： 1. iCode9 技术分享网（下文简称本站）提供的所有内容，仅供技术学习、探讨和分享；
2. 关于本站的所有留言、评论、转载及引用，纯属内容发起人的个人观点，与本站观点和立场无关；
3. 关于本站的所有言论和文字，纯属内容发起人的个人观点，与本站观点和立场无关；
4. 本站文章均是网友提供，不完全保证技术分享内容的完整性、准确性、时效性、风险性和版权归属；如您发现该文章侵犯了您的权益，可联系我们第一时间进行删除；
5. 本站为非盈利性的个人网站，所有内容不会用来进行牟利，也不会利用任何形式的广告来间接获益，纯粹是为了广大技术爱好者提供技术内容和技术思想的分享性交流网站。