如何在rllib中输入图像

Question

上次我看到库rllib：https://docs.ray.io/en/latest/rllib/index.html。它有令人惊奇的强化学习功能，但不幸的是，我无法找到一种方法来输入图像作为一个观察，而不平坦(我基本上想使用卷积神经网络)。有没有任何方法可以使用rllib库在模型中输入图像观察？

Answer 1

Rllib与openai的健身房兼容，您可以创建一个定制的env https://docs.ray.io/en/latest/rllib/rllib-env.html#configuring-environments，并将一个Box作为像https://stackoverflow.com/a/69602365/4994352这样的观察空间返回