Evaluate on Reasoning Tasks

[JW-Yuan]

In the function calculate_metric(args, target, ours),there is a

assert colors.shape[0] == 2, colors # white and the expected color

and I always got an error

File "/home/featurize/work/yjw/visual_prompting/evaluate/evaluate_reasoning.py", line 129, in calculate_metric assert colors.shape[0] == 2, colors # white and the expected color AssertionError: [[ 0 0 255] [ 0 255 0] [255 255 255]]

which means I've got three colors, I guess it is related to the issues about functions transform or round_image.

Did you encounter this promblem? Looking forward to your reply.