Commit 689805b9 authored by YaningGao's avatar YaningGao
Browse files

update svg score logic

parent 98041a52

vagen/env/README.md

+3 −0
Original line number Diff line number Diff line
@@ -44,6 +44,9 @@ pip install flask 
# create server for reward model

python vagen/env/svgdino/reward_model_server.py



# test whether reward model server functions well

python vagen/examples/debug_reward_model_server/reward_server_benchmark.py --url http://127.0.0.1:5000 --concurrency 128 --requests 1000



# Then run experiment of SVG simply copy the code below

bash vagen/examples/debug_svgdino_vision_grpo/run.sh

```
 
 No newline at end of file

vagen/env/svg/score.py

+1 −0
Original line number Diff line number Diff line
@@ -71,6 +71,7 @@ def calculate_total_score(gt_im, gen_im, gt_code, gen_code, score_config): 
    model_size = score_config.get("model_size", "large")

    dino_only = score_config.get("dino_only", False)



    if score_config.dino_weight != 0:

        reward_model = DINOScoreCalculator(model_size=model_size, device="cuda")

        dino_score = reward_model.calculate_DINOv2_similarity_score(gt_im=gt_im, gen_im=gen_im)

CRA Git | Maintained and supported by SUSTech CRA and CCSE