File size: 2,120 Bytes
28c0435 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 |
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)
|