About loading CLIPScore and BLIPScore

THUDM / ImageReward

[NeurIPS 2023] ImageReward: Learning and Evaluating Human Preferences for Text-to-image Generation

Apache License 2.0

1.12k stars 62 forks source link

About loading CLIPScore and BLIPScore #69

Closed passing2961 closed 10 months ago

passing2961 commented 10 months ago

Hi,

Is it correct to load the CLIPScore and BLIPScore to reproduce the results in your paper? Check the below codes.

# load BLIPScore
blipscore = RM.load_score("BLIP")
blipscore.score(prompt, image_file_dir)

# load CLIPScore
clipscore = RM.load_score("CLIP")
clipscore.score(prompt, image_file_dir)

# load ImageReward
image_reward = RM.load("ImageReward-v1.0")
image_reward.score(prompt, image_file_dir)

Best,

xujz18 commented 10 months ago

You can refer to test.py