# ann_file="/data/dataset/dataset_json/data/flickr30k_train.json"
# output_file="/data/dataset/dataset_json/data_rewrite/flickr30k_train.json"

# ann_file="/data/dataset/dataset_json/data_rewrite/flickr30k_train_llama_bard.json"
# output_file="/data/dataset/dataset_json/data_rewrite/flickr30k_train_llama_bard_limit.json"

# ann_file="/data/dataset/dataset_json/data_rewrite/flickr30k_train_llama_coco_multi_top1.json"
# output_file="/data/dataset/dataset_json/data_rewrite/flickr30k_train_llama_coco_multi_top1_limit.json"

# ann_file="/data/dataset/dataset_json/data_rewrite/flickr30k_train_llama_coco_multi_top3.json"
# output_file="/data/dataset/dataset_json/data_rewrite/flickr30k_train_llama_coco_multi_top3_limit.json"

ann_file="/data/dataset/dataset_json/data_rewrite/flickr30k_train_llama_coco_multi_idx1.json"
output_file="/data/dataset/dataset_json/data_rewrite/flickr30k_train_llama_coco_multi_idx1_limit.json"

python3 limit_train_file_caps.py \
    --ann_file ${ann_file} \
    --out_file ${output_file}
