| import argparse | |
| import os | |
| import pandas as pd | |
| # ['ALIGN:align-basecoyo700m', 'CLIPS:CLIPS-Large-14-224recap-datacomp1b', 'CLIPS:CLIPS-Large-14-336recap-datacomp1b', 'CLIPS:CLIPS-Huge-14-224recap-datacomp1b', 'DreamLIP:dreamlip-vitb16cc3m-long', 'DreamLIP:dreamlip-vitb16cc12m-long', 'DreamLIP:dreamlip-vitb16yfcc15m-long', 'DreamLIP:dreamlip-vitb16cc30m-long', 'FSC-CLIP:fsc-clip-ViT-B-32laioncoco-ft', 'FSC-CLIP:fsc-clip-ViT-B-16laioncoco-ft', 'FSC-CLIP:fsc-clip-ViT-L-14laioncoco-ft', 'Jina-CLIP:jina-clip-v1jinaai', 'Jina-CLIP:jina-clip-v2jinaai', 'LoTLIP:LoTLIP-ViT-B-32lotlip100m', 'LoTLIP:LoTLIP-ViT-B-16lotlip100m', 'LongCLIP:longclip-vitb32sharegpt4v-1m', 'LongCLIP:longclip-vitb16sharegpt4v-1m', 'LongCLIP:longclip-vitl14sharegpt4v-1m', 'LongCLIP:longclip-vitl14_336pxsharegpt4v-1m', 'OpenCLIP:roberta-ViT-B-32laion2b_s12b_b32k', 'OpenCLIP:coca_ViT-B-32laion2b_s13b_b90k', 'OpenCLIP:coca_ViT-L-14laion2b_s13b_b90k', 'OpenCLIP:ViT-H-14laion2b_s32b_b79k', 'OpenCLIP:ViT-L-14laion2b_s32b_b82k', 'OpenCLIP:ViT-B-32laion2b_s34b_b79k', 'OpenCLIP:ViT-B-16laion2b_s34b_b88k', 'OpenCLIP:ViT-g-14laion2b_s34b_b88k', 'OpenCLIP:ViT-B-16openai', 'OpenCLIP:ViT-B-32openai', 'OpenCLIP:ViT-L-14openai', 'OpenCLIP:ViT-L-14-336openai', 'OpenCLIP:ViT-B-16-SigLIPwebli', 'OpenCLIP:ViT-B-16-SigLIP-384webli', 'OpenCLIP:ViT-L-16-SigLIP-256webli', 'OpenCLIP:ViT-L-16-SigLIP-384webli', 'OpenCLIP:ViT-SO400M-14-SigLIPwebli', 'Recap-CLIP:ViT-L-16-HTxt-Recap-CLIPrecap-datacomp1b', 'StructuredCLIP:NegCLIP-ViT-B-32coco-ft', 'StructuredCLIP:CE-CLIP-ViT-B-32coco-ft', 'StructuredCLIP:DAC-LLM-ViT-B-32cc3m-ft', 'StructuredCLIP:DAC-SAM-ViT-B-32cc3m-ft'] | |
| def parse_args(): | |
| parser = argparse.ArgumentParser() | |
| parser.add_argument("--csv-file", type=str) | |
| return parser.parse_args() | |
| def main(args): | |
| df = pd.read_csv(args.csv_file) | |
| model_keys = [] | |
| for _, row in df.iterrows(): | |
| model_key = row["family"] + ":" + row["model"] + ":" + row["tag"] | |
| model_keys.append(model_key) | |
| import pdb | |
| pdb.set_trace() | |
| if __name__ == "__main__": | |
| args = parse_args() | |
| main(args) | |