chendl's picture
Add application file
0b7b08a
raw
history blame
591 Bytes
import os
import json
if __name__ == "__main__":
blip2_cases = os.listdir("/gpfs/u/home/LMCG/LMCGljnn/scratch/code/multimodal2/blip2_baseline/blip2_fail_case")
kmos2_cases = os.listdir("/gpfs/u/home/LMCG/LMCGljnn/scratch/code/unilm/kosmos-2/kmos2_fail_case")
blip2_failed_ids = set([int(c.split("_")[0]) for c in blip2_cases])
kmos2_failed_ids = set([int(c.split("_")[0]) for c in kmos2_cases])
both_failed_ids = list(blip2_failed_ids.intersection(kmos2_failed_ids))
print(both_failed_ids)
json.dump(both_failed_ids, open("both_failed_ids.json", "w"), indent=1)