Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

[REQUEST] Enable observation dictionary input. #249

Open
Emiyalzn opened this issue Dec 15, 2022 · 3 comments
Open

[REQUEST] Enable observation dictionary input. #249

Emiyalzn opened this issue Dec 15, 2022 · 3 comments
Labels
enhancement New feature or request

Comments

@Emiyalzn
Copy link

Emiyalzn commented Dec 15, 2022

Is your feature request related to a problem? Please describe.
Currently, your MDPDataset class assert the observation to be an ndarray object. However, In the field of autonomous driving, the MDP observation cannot be represented by a simple ndarray object. Typically, the observation space can be composed of a BEV image and a speed profile, which is not supported by your MDPDataset yet.

Describe the solution you'd like
I believe it will make the repo stronger to enable observation dictionary storage and training like {"BEV": ndarray(C, W, H), "speed": (1,)} in the MDPDataset (as well as Episode and Transition class).

@Emiyalzn Emiyalzn added the enhancement New feature or request label Dec 15, 2022
@takuseno
Copy link
Owner

takuseno commented Jan 9, 2023

@Emiyalzn Thanks for the issue and sorry for the late response. I'm currently pushed by my Ph.D thesis. But, I'm actually working on tuple observation support in nightly branch. Hope this solves your problem once it's released.

@Emiyalzn
Copy link
Author

Thank you. I use the rllib to solve my problem, and maybe you can refer to their implementation for some help.

@takuseno
Copy link
Owner

Good to know you have a working code now. No worry, nightly branch already works now. But, I'm still making more changes until release.

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
enhancement New feature or request
Projects
None yet
Development

No branches or pull requests

2 participants