Invalid line while taking target_image in Data_util.py

Hello,
I found a line that seems weird and share this information to check whether I got some misunderstanding or should be fixed.

As far as i know, the agent_id should be 0 or 1, so it cannot be reachable to 500.
And from this line, I think the code should be change to "action_id" (because action ID 501 and 502 are about the searching API)

teach_tatc/src/teach/modeling/utils/data_util.py

Line 48 in e19fef1

if interaction["agent_id"] > 500 and interaction["success"]:

And also, The target_idx (variable "i" in below) also seems not valid. (I just fixed it for my own but for information)

teach_tatc/src/teach/modeling/utils/data_util.py

Line 51 in e19fef1

target_idx.append(interactions.index(i))

Thanks!

Thanks for pointing out! You are right, those variables should be "agent_id"-->"action_id" and i-->interaction.

Thank you for your reply! BTW, I have another question.

As the explanation in the paper, it seems we could get the ground truth of segmentation for target object.

But in the dataset I downloaded looks like below unlike the paper.

I think something goes wrong and try to download the dataset again but seems hard to redownload it.
(Downloading meta-data from google drive always fail due to network error)

Is the dataset I got wrong? And if so, could you please check whether the meta_data.tar.gz file could be downloaded normally?

Thanks!

Thanks for bringing this up! I have pushed a fix for this bug now, and the segmentation masks should render correctly on replay. Let me know if the bug persists after pulling 634c011 and how to produce it if so.