allenai / ai2thor

An open-source platform for Visual AI.
http://ai2thor.allenai.org
Apache License 2.0
1.15k stars 215 forks source link

color labels in robothor #1141

Open AnukritiSinghh opened 1 year ago

AnukritiSinghh commented 1 year ago

I added a third party camera and displayed the segmented images, the images look correct but once I do np.unique or use set, I am not able to extract the labels of the segmented image. It prints a large number like 34, even if the labels look like hardly 6. Am I doing something wrong? I have tried and this work correctly if the segmented images are taken from the agent, but only fails when they are taken from third party camera. Please help

image