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
 
                    
                     
                    
                 
                    
                
 
                
            
         
         浙公网安备 33010602011771号
浙公网安备 33010602011771号