INNER CODE UNIT · Python
get_winoground_acc
linzhiqiu/t2v_metrics · dataset.py:203
def get_winoground_acc(scores):
text_correct_count = 0
image_correct_count = 0
group_correct_count = 0
def text_correct(result):
return result["c0_i0"] > result["c1_i0"] and result["c1_i1"] > result["c0_i1"]
def image_correct(result):
return result["c0_i0"] > result["c0_i1"] and result["c1_i1"] > result["c1_i0"]
def group_correct(result):
return image_correct(result) and text_correct(result)
for result in scores:
text_correct_count += 1 if text_correct(result) else 0
image_correct_count += 1 if image_correct(result) else 0
group_correct_count += 1 if group_correct(result) else 0