CoOp

CoOp训练

  • 1.准备训练数据,按照每个类别分文件夹存放
  • 2.datasets增加数据定义文件
    CoOp-main\configs\datasets
  • 3.增加数据加载处理文件,可参考dtd.py
  • 4.修改CoOp-main\scripts\coop\main.sh中的
    DATA数据目录
    DIR工作目录
  • 5.CoOp-main\train.py中 import步骤2定义的数据文件
  • 6.如果不方便加载clip模型,改为本地加载模型CoOp-main\trainers\coop.py中load_clip_to_cpu方法的model_path=本地模型路径

如果是需要测试0-shot能力

  • 1、增加新数据的模板CoOp-main\trainers\zsclip.py中CUSTOM_TEMPLATES
posted @ 2022-08-30 17:53  哈哈哈喽喽喽  阅读(248)  评论(0编辑  收藏  举报