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.

Sign up for GitHub

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

Jump to bottom

How to get the rgb images masks #64

Open

LogSSim opened this issue Oct 16, 2024 · 1 comment

LogSSim commented Oct 16, 2024

I saw an observation of RLbench including the masks of RGB images. How could I get it in this code?

Collaborator

imankgoyal commented Oct 18, 2024

Hi,

We use dataloader functions from PerAct. You might have to modify them ground up to make masks available. Some places you can start looking at:

-- https://github.com/NVlabs/RVT/blob/master/rvt/utils/dataset.py#L262-L268
-- https://github.com/NVlabs/peract/blob/5c2988edb961d67d7a921cbbc638f69947debff8/helpers/utils.py#L322

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment