affordance prediction with pretrained model

[Tom-Huang]

@SudeepDasari Dear authors of the paper, thank you so much for your great work and the release of the code. I wonder if it is possible to use the pre-trained model to predict affordance from the image like in the VRB paper. Could you also provide the code and weight for that? Thank you so much!

[mohanksriram]

Hi Tom, Please use this script on the hrp_release branch for attention visualization!

Example usage:

python examples/explore_attention_maps.py --vit_model_name={VC1_hrp|SOUP_1M_DH|Any vit_base/hrp model} --img_path={path/to/image.png}