强化学习:仿真环境Gym的安装和测试

接下来进行强化学习方面的研究和学习:
首先是安装学习RL最常用的仿真环境——Gym

推荐安装的Gym版本如下,使用pip命令进行安装:

pip install gym==0.26.2

pip install pygame==2.3.0

环境安装完成后,可以使用下面的测试代码进行测试:

# RL环境Gym的测试代码
import gym

# 创建CartPole环境,指定渲染模式为rgb_array,如果是在IDE中可以改为'human’
env = gym.make('CartPole-v1', render_mode='human')

# 重置环境
env.reset()

# 循环N次
for i in range(10):
    env.render()  # 渲染环境
    action = env.action_space.sample()  # 从动作空间中随机选取一个动作
    env.step(action)  # 执行动作

# 关闭环境
env.close()

我是在Pycharm编辑器中进行的环境测试,结果如下:

运行测试代码时会发现会有一个倒立摆短时出现,说明Gym环境安装的测试无误!

最后给出结合opencv实现gym环境的图像动态显示代码案例:

import cv2
import gym
import numpy as np

class GymHelper:
    def __init__(self, env):
        self.env = env
        # OpenCV不需要在这里初始化图像窗口

    def render(self):
        # 在新的API中,确保你在创建环境时指定了render_mode='rgb_array'
        image_data = self.env.render()  # 获取RGB数组
        image_data = cv2.cvtColor(image_data, cv2.COLOR_RGB2BGR)  # 将RGB转为BGR
        cv2.imshow('Environment', image_data)  # 使用OpenCV显示图像
        cv2.waitKey(100)  # 短暂等待,以便图像可以显示

# 创建环境,指定渲染模式
env = gym.make('CartPole-v1', render_mode='rgb_array')
env.reset()
gym_helper = GymHelper(env)

for i in range(30):
    gym_helper.render()  # 使用OpenCV渲染环境
    action = env.action_space.sample()
    env.step(action)

# 循环结束后关闭OpenCV窗口
cv2.destroyAllWindows()
env.close()

运行结果如下,图像是有一段动态过程的,我只进行了一次静态截图显示:

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包

打赏作者

随机惯性粒子群

你的鼓励将是我创作的最大动力

¥1 ¥2 ¥4 ¥6 ¥10 ¥20
扫码支付:¥1
获取中
扫码支付

您的余额不足,请更换扫码支付或充值

打赏作者

实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值