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