openai/gym中的图像预处理
openai/gym中的图像预处理
之前读强化学习的文章,提到对gym模拟器的图像进行预处理,每4张图片生成一个(84,84,4)的tensor,但是在网上没有搜到具体的实现,因此写一个预处理的函数,用到了cv2,这个模块需要安装opencv-python这个库
代码如下:
import matplotlib.pyplot as plt
import gym
import numpy as np
...
原创
2018-12-15 14:58:28 ·
1919 阅读 ·
2 评论