askforalfred / alfred

ALFRED - A Benchmark for Interpreting Grounded Instructions for Everyday Tasks
MIT License
352 stars 77 forks source link

augment_trajectories.py throws error regarding Original Image Count and New Image Count #113

Open jesbu1 opened 2 years ago

jesbu1 commented 2 years ago

When cloning the repo from scratch and using either just the json feats or the full dataset, running augment_trajectories.py with no changes throws this error for me for all trajectories:

Original Image Count 514, New Image Count 0
Traceback (most recent call last):
  File "/home/jesse/alfred/gen/scripts/augment_trajectories.py", line 264, in run
    augment_traj(env, json_file)
  File "/home/jesse/alfred/gen/scripts/augment_trajectories.py", line 244, in augment_traj
    raise Exception("WARNING: the augmented sequence length doesn't match the original")
Exception: WARNING: the augmented sequence length doesn't match the original
Error: Exception("WARNING: the augmented sequence length doesn't match the original")
MohitShridhar commented 2 years ago

@jesbu1, this is a bit strange. It looks like no new images are being saved. Do you see some agents running around in ai2thor windows? Is anything being saved at all?

jesbu1 commented 2 years ago

I ended up finding the isseu: what happened is the raw_images are in .jpg and the function that searches for images only looks at .pngs

The images were still saved regardless

jesbu1 commented 2 years ago

I can make a PR in a bit to fix this bug!