Rain-Bus/PromptSRC
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

rename to dzgcoop

Browse Source
This commit is contained in:
Rain&Bus
2026-02-24 01:01:53 +08:00
parent f3a7993665
commit 85c6591232
13 changed files with 73 additions and 73 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
configs/trainers/PromptSRC/vit_b16_c2_ep20_batch4_4+4ctx.yaml → configs/trainers/DZGCoOp/vit_b16_c2_ep20_batch4_4+4ctx.yaml
View File

@@ -1,4 +1,4 @@
# PromptSRC: Prompting with Self-regularizing constraints # DZGCoOp: Dual-branch Zero-shot Guidance CoOp
DATALOADER: DATALOADER:
TRAIN_X: TRAIN_X:
BATCH_SIZE: 4 BATCH_SIZE: 4
@@ -30,7 +30,7 @@ MODEL:
NAME: "ViT-B/16" NAME: "ViT-B/16"
TRAINER: TRAINER:
PROMPTSRC: DZGCOOP:
N_CTX_VISION: 4 N_CTX_VISION: 4
N_CTX_TEXT: 4 N_CTX_TEXT: 4
CTX_INIT: "a photo of a" CTX_INIT: "a photo of a"

4
configs/trainers/PromptSRC/vit_b16_c2_ep20_batch4_4+4ctx_cross_datasets.yaml → configs/trainers/DZGCoOp/vit_b16_c2_ep20_batch4_4+4ctx_cross_datasets.yaml
View File

@@ -1,4 +1,4 @@
# PromptSRC: Prompting with Self-regularizing constraints # DZGCoOp: Dual-branch Zero-shot Guidance CoOp
DATALOADER: DATALOADER:
TRAIN_X: TRAIN_X:
BATCH_SIZE: 4 BATCH_SIZE: 4
@@ -31,7 +31,7 @@ MODEL:
NAME: "ViT-B/16" NAME: "ViT-B/16"
TRAINER: TRAINER:
PROMPTSRC: DZGCOOP:
N_CTX_VISION: 4 N_CTX_VISION: 4
N_CTX_TEXT: 4 N_CTX_TEXT: 4
CTX_INIT: "a photo of a" CTX_INIT: "a photo of a"

4
configs/trainers/PromptSRC/vit_b16_c2_ep5_batch4_4+4ctx_cross_datasets.yaml → configs/trainers/DZGCoOp/vit_b16_c2_ep5_batch4_4+4ctx_cross_datasets.yaml
View File

@@ -1,4 +1,4 @@
# PromptSRC: Prompting with Self-regularizing constraints # DZGCoOp: Dual-branch Zero-shot Guidance CoOp
DATALOADER: DATALOADER:
TRAIN_X: TRAIN_X:
BATCH_SIZE: 4 BATCH_SIZE: 4
@@ -30,7 +30,7 @@ MODEL:
NAME: "ViT-B/16" NAME: "ViT-B/16"
TRAINER: TRAINER:
PROMPTSRC: DZGCOOP:
N_CTX_VISION: 4 N_CTX_VISION: 4
N_CTX_TEXT: 4 N_CTX_TEXT: 4
CTX_INIT: "a photo of a" CTX_INIT: "a photo of a"

2
extract_acc.py
View File

@@ -109,7 +109,7 @@ def print_model_results(results, model_name):
def main(): def main():
root_dir = 'output' # 修改为你的output目录路径 root_dir = 'output' # 修改为你的output目录路径
target_model = 'PromptSRC' # 指定要分析的模型 target_model = 'DZGCoOp' # 指定要分析的模型
results = collect_model_results(root_dir, target_model) results = collect_model_results(root_dir, target_model)
print_model_results(results, target_model) print_model_results(results, target_model)

22
scripts/dzgcoop/base2new_all.sh Normal file
View File

@@ -0,0 +1,22 @@
seeds=(1 2 3)
datasets=(
"ucf101"
"eurosat"
"oxford_pets"
"food101"
"oxford_flowers"
"dtd"
"caltech101"
"fgvc_aircraft"
"stanford_cars"
# "sun397"
# "imagenet"
)
for dataset in "${datasets[@]}"; do
for seed in "${seeds[@]}"; do
bash scripts/dzgcoop/base2new_train.sh "$dataset" "$seed"
bash scripts/dzgcoop/base2new_test.sh "$dataset" "$seed"
done
done

2
scripts/promptsrc/base2new_test.sh → scripts/dzgcoop/base2new_test.sh
View File

@@ -3,7 +3,7 @@
# custom config # custom config
DATA="~/Datasets/CoOp" DATA="~/Datasets/CoOp"
TRAINER=PromptSRC TRAINER=DZGCoOp
DATASET=$1 DATASET=$1
SEED=$2 SEED=$2

2
scripts/promptsrc/base2new_train.sh → scripts/dzgcoop/base2new_train.sh
View File

@@ -2,7 +2,7 @@
# custom config # custom config
DATA="~/Datasets/CoOp" DATA="~/Datasets/CoOp"
TRAINER=PromptSRC TRAINER=DZGCoOp
DATASET=$1 DATASET=$1
SEED=$2 SEED=$2

2
scripts/promptsrc/xd_train.sh → scripts/dzgcoop/xd_train.sh
View File

@@ -2,7 +2,7 @@
DATA=" ~/Datasets/CoOp" DATA=" ~/Datasets/CoOp"
TRAINER=PromptSRC TRAINER=DZGCoOp
SRC_DATASETS=imagenet SRC_DATASETS=imagenet
SHOTS=16 SHOTS=16
CFG=vit_b16_c2_ep20_batch4_4+4ctx_cross_datasets CFG=vit_b16_c2_ep20_batch4_4+4ctx_cross_datasets

2
scripts/promptsrc/xda_test.sh → scripts/dzgcoop/xda_test.sh
View File

@@ -3,7 +3,7 @@
# custom config # custom config
DATA=" ~/Datasets/CoOp" DATA=" ~/Datasets/CoOp"
TRAINER=PromptSRC TRAINER=DZGCoOp
SRC_DATASETS=imagenet SRC_DATASETS=imagenet

2
scripts/promptsrc/xdo_test.sh → scripts/dzgcoop/xdo_test.sh
View File

@@ -3,7 +3,7 @@
# custom config # custom config
DATA=" ~/Datasets/CoOp" DATA=" ~/Datasets/CoOp"
TRAINER=PromptSRC TRAINER=DZGCoOp
SRC_DATASETS=imagenet SRC_DATASETS=imagenet

22
scripts/promptsrc/base2new_all.sh
View File

@@ -1,22 +0,0 @@
seeds=(1 2 3)
datasets=(
# "ucf101"
# "eurosat"
# "oxford_pets"
# "food101"
# "oxford_flowers"
# "dtd"
# "caltech101"
# "fgvc_aircraft"
# "stanford_cars"
# "sun397"
"imagenet"
)
for dataset in "${datasets[@]}"; do
for seed in "${seeds[@]}"; do
bash scripts/promptsrc/base2new_train.sh "$dataset" "$seed"
bash scripts/promptsrc/base2new_test.sh "$dataset" "$seed"
done
done

30
train.py
View File

@@ -28,7 +28,7 @@ import trainers.cocoop
import trainers.zsclip import trainers.zsclip
import trainers.maple import trainers.maple
import trainers.independentVL import trainers.independentVL
import trainers.promptsrc import trainers.dzgcoop
def print_args(args, cfg): def print_args(args, cfg):
@@ -110,20 +110,20 @@ def extend_cfg(cfg):
cfg.TRAINER.MAPLE.PROMPT_DEPTH = 9 # Max 12, minimum 0, for 1 it will act as shallow MaPLe (J=1) cfg.TRAINER.MAPLE.PROMPT_DEPTH = 9 # Max 12, minimum 0, for 1 it will act as shallow MaPLe (J=1)
cfg.DATASET.SUBSAMPLE_CLASSES = "all" # all, base or new cfg.DATASET.SUBSAMPLE_CLASSES = "all" # all, base or new
# Config for PromptSRC # Config for DZGCoOp
cfg.TRAINER.PROMPTSRC = CN() cfg.TRAINER.DZGCOOP = CN()
cfg.TRAINER.PROMPTSRC.N_CTX_VISION = 4 # number of context vectors at the vision branch cfg.TRAINER.DZGCOOP.N_CTX_VISION = 4 # number of context vectors at the vision branch
cfg.TRAINER.PROMPTSRC.N_CTX_TEXT = 4 # number of context vectors at the language branch cfg.TRAINER.DZGCOOP.N_CTX_TEXT = 4 # number of context vectors at the language branch
cfg.TRAINER.PROMPTSRC.CTX_INIT = "a photo of a" # initialization words cfg.TRAINER.DZGCOOP.CTX_INIT = "a photo of a" # initialization words
cfg.TRAINER.PROMPTSRC.PREC = "fp16" # fp16, fp32, amp cfg.TRAINER.DZGCOOP.PREC = "fp16" # fp16, fp32, amp
cfg.TRAINER.PROMPTSRC.PROMPT_DEPTH_VISION = 9 # Max 12, minimum 0, for 0 it will be using shallow IVLP prompting (J=1) cfg.TRAINER.DZGCOOP.PROMPT_DEPTH_VISION = 9 # Max 12, minimum 0, for 0 it will be using shallow IVLP prompting (J=1)
cfg.TRAINER.PROMPTSRC.PROMPT_DEPTH_TEXT = 9 # Max 12, minimum 0, for 0 it will be using shallow IVLP prompting (J=1) cfg.TRAINER.DZGCOOP.PROMPT_DEPTH_TEXT = 9 # Max 12, minimum 0, for 0 it will be using shallow IVLP prompting (J=1)
cfg.TRAINER.PROMPTSRC.TEXT_LOSS_WEIGHT = 25 cfg.TRAINER.DZGCOOP.TEXT_LOSS_WEIGHT = 25
cfg.TRAINER.PROMPTSRC.TEXT_LOSS_WEIGHT_STRONG = 25 # lambda2: strong text constraint weight cfg.TRAINER.DZGCOOP.TEXT_LOSS_WEIGHT_STRONG = 25 # lambda2: strong text constraint weight
cfg.TRAINER.PROMPTSRC.TEXT_LOSS_WEIGHT_WEAK = 2.5 # lambda3: weak text constraint weight cfg.TRAINER.DZGCOOP.TEXT_LOSS_WEIGHT_WEAK = 2.5 # lambda3: weak text constraint weight
cfg.TRAINER.PROMPTSRC.IMAGE_LOSS_WEIGHT = 10 cfg.TRAINER.DZGCOOP.IMAGE_LOSS_WEIGHT = 10
cfg.TRAINER.PROMPTSRC.GPA_MEAN = 15 cfg.TRAINER.DZGCOOP.GPA_MEAN = 15
cfg.TRAINER.PROMPTSRC.GPA_STD = 1 cfg.TRAINER.DZGCOOP.GPA_STD = 1
cfg.DATASET.SUBSAMPLE_CLASSES = "all" # all, base or new cfg.DATASET.SUBSAMPLE_CLASSES = "all" # all, base or new
# Config for independent Vision Language prompting (independent-vlp) # Config for independent Vision Language prompting (independent-vlp)

48
trainers/promptsrc.py → trainers/dzgcoop.py
View File

@@ -51,10 +51,10 @@ def load_clip_to_cpu(cfg, zero_shot_model=False):
state_dict = torch.load(model_path, map_location="cpu") state_dict = torch.load(model_path, map_location="cpu")
if not zero_shot_model: if not zero_shot_model:
design_details = {"trainer": 'IVLP', design_details = {"trainer": 'IVLP',
"vision_depth": cfg.TRAINER.PROMPTSRC.PROMPT_DEPTH_VISION, "vision_depth": cfg.TRAINER.DZGCOOP.PROMPT_DEPTH_VISION,
"language_depth": cfg.TRAINER.PROMPTSRC.PROMPT_DEPTH_TEXT, "language_depth": cfg.TRAINER.DZGCOOP.PROMPT_DEPTH_TEXT,
"vision_ctx": cfg.TRAINER.PROMPTSRC.N_CTX_VISION, "vision_ctx": cfg.TRAINER.DZGCOOP.N_CTX_VISION,
"language_ctx": cfg.TRAINER.PROMPTSRC.N_CTX_TEXT} "language_ctx": cfg.TRAINER.DZGCOOP.N_CTX_TEXT}
model = clip.build_model(state_dict or model.state_dict(), design_details) model = clip.build_model(state_dict or model.state_dict(), design_details)
else: else:
# Return original CLIP model for generating frozen VL features # Return original CLIP model for generating frozen VL features
@@ -95,11 +95,11 @@ class VLPromptLearner(nn.Module):
super().__init__() super().__init__()
n_cls = len(classnames) n_cls = len(classnames)
# Make sure Language depth >= 1 # Make sure Language depth >= 1
assert cfg.TRAINER.PROMPTSRC.PROMPT_DEPTH_TEXT >= 1, "In Independent VL prompting, Language prompt depth should be >=1" \ assert cfg.TRAINER.DZGCOOP.PROMPT_DEPTH_TEXT >= 1, "In Independent VL prompting, Language prompt depth should be >=1" \
"\nPlease use VPT trainer if you want to learn only vision " \ "\nPlease use VPT trainer if you want to learn only vision " \
"branch" "branch"
n_ctx = cfg.TRAINER.PROMPTSRC.N_CTX_TEXT n_ctx = cfg.TRAINER.DZGCOOP.N_CTX_TEXT
ctx_init = cfg.TRAINER.PROMPTSRC.CTX_INIT ctx_init = cfg.TRAINER.DZGCOOP.CTX_INIT
dtype = clip_model.dtype dtype = clip_model.dtype
ctx_dim = clip_model.ln_final.weight.shape[0] ctx_dim = clip_model.ln_final.weight.shape[0]
clip_imsize = clip_model.visual.input_resolution clip_imsize = clip_model.visual.input_resolution
@@ -126,7 +126,7 @@ class VLPromptLearner(nn.Module):
print(f'Strong branch initial text context: "{prompt_prefix_strong}"') print(f'Strong branch initial text context: "{prompt_prefix_strong}"')
print(f'Weak branch initial text context: "{prompt_prefix_weak}"') print(f'Weak branch initial text context: "{prompt_prefix_weak}"')
print(f"Number of context words (tokens) for Language prompting: {n_ctx}") print(f"Number of context words (tokens) for Language prompting: {n_ctx}")
print(f"Number of context words (tokens) for Vision prompting: {cfg.TRAINER.PROMPTSRC.N_CTX_VISION}") print(f"Number of context words (tokens) for Vision prompting: {cfg.TRAINER.DZGCOOP.N_CTX_VISION}")
self.ctx_strong = nn.Parameter(ctx_vectors_strong) self.ctx_strong = nn.Parameter(ctx_vectors_strong)
self.ctx_weak = nn.Parameter(ctx_vectors_weak) self.ctx_weak = nn.Parameter(ctx_vectors_weak)
@@ -261,9 +261,9 @@ class CustomCLIP(nn.Module):
@TRAINER_REGISTRY.register() @TRAINER_REGISTRY.register()
class PromptSRC(TrainerX): class DZGCoOp(TrainerX):
def check_cfg(self, cfg): def check_cfg(self, cfg):
assert cfg.TRAINER.PROMPTSRC.PREC in ["fp16", "fp32", "amp"] assert cfg.TRAINER.DZGCOOP.PREC in ["fp16", "fp32", "amp"]
def build_model(self): def build_model(self):
cfg = self.cfg cfg = self.cfg
@@ -272,7 +272,7 @@ class PromptSRC(TrainerX):
print(f"Loading CLIP (backbone: {cfg.MODEL.BACKBONE.NAME})") print(f"Loading CLIP (backbone: {cfg.MODEL.BACKBONE.NAME})")
clip_model = load_clip_to_cpu(cfg) clip_model = load_clip_to_cpu(cfg)
if cfg.TRAINER.PROMPTSRC.PREC == "fp32" or cfg.TRAINER.PROMPTSRC.PREC == "amp": if cfg.TRAINER.DZGCOOP.PREC == "fp32" or cfg.TRAINER.DZGCOOP.PREC == "amp":
# CLIP's default precision is fp16 # CLIP's default precision is fp16
clip_model.float() clip_model.float()
@@ -312,12 +312,12 @@ class PromptSRC(TrainerX):
self.total_epochs = cfg.OPTIM.MAX_EPOCH self.total_epochs = cfg.OPTIM.MAX_EPOCH
self.step_counter = 1 self.step_counter = 1
N = cfg.OPTIM.MAX_EPOCH N = cfg.OPTIM.MAX_EPOCH
mean = cfg.TRAINER.PROMPTSRC.GPA_MEAN mean = cfg.TRAINER.DZGCOOP.GPA_MEAN
stdev = cfg.TRAINER.PROMPTSRC.GPA_STD stdev = cfg.TRAINER.DZGCOOP.GPA_STD
gauss = self.get_gauss(mean, stdev) gauss = self.get_gauss(mean, stdev)
self.gauss = np.array([gauss(a) for a in range(1, N + 1)]) self.gauss = np.array([gauss(a) for a in range(1, N + 1)])
self.gauss = self.gauss / sum(self.gauss) self.gauss = self.gauss / sum(self.gauss)
self.scaler = GradScaler() if cfg.TRAINER.PROMPTSRC.PREC == "amp" else None self.scaler = GradScaler() if cfg.TRAINER.DZGCOOP.PREC == "amp" else None
# Note that multi-gpu training could be slow because CLIP's size is # Note that multi-gpu training could be slow because CLIP's size is
# big, which slows down the copy operation in DataParallel # big, which slows down the copy operation in DataParallel
device_count = torch.cuda.device_count() device_count = torch.cuda.device_count()
@@ -334,7 +334,7 @@ class PromptSRC(TrainerX):
optim = self.optim optim = self.optim
scaler = self.scaler scaler = self.scaler
prec = self.cfg.TRAINER.PROMPTSRC.PREC prec = self.cfg.TRAINER.DZGCOOP.PREC
if prec == "amp": if prec == "amp":
with autocast(): with autocast():
loss = model(image, label) loss = model(image, label)
@@ -346,23 +346,23 @@ class PromptSRC(TrainerX):
loss_ce, text_features_strong, text_features_weak, fixed_embeddings, zs_image_embedd, image_ft, \ loss_ce, text_features_strong, text_features_weak, fixed_embeddings, zs_image_embedd, image_ft, \
zero_shot_logits, logits_strong, logits_weak, logits_final = model(image, label) zero_shot_logits, logits_strong, logits_weak, logits_final = model(image, label)
lambda1 = self.cfg.TRAINER.PROMPTSRC.IMAGE_LOSS_WEIGHT lambda1 = self.cfg.TRAINER.DZGCOOP.IMAGE_LOSS_WEIGHT
lambda2 = self.cfg.TRAINER.PROMPTSRC.TEXT_LOSS_WEIGHT_STRONG lambda2 = self.cfg.TRAINER.DZGCOOP.TEXT_LOSS_WEIGHT_STRONG
lambda3 = self.cfg.TRAINER.PROMPTSRC.TEXT_LOSS_WEIGHT_WEAK lambda3 = self.cfg.TRAINER.DZGCOOP.TEXT_LOSS_WEIGHT_WEAK
loss_scl_image = F.l1_loss(image_ft, zs_image_embedd.cuda(), reduction='mean') * lambda1 L_zvg = F.l1_loss(image_ft, zs_image_embedd.cuda(), reduction='mean') * lambda1
loss_scl_text_strong = F.l1_loss(text_features_strong, fixed_embeddings.cuda(), reduction='mean') * lambda2 L_sg_strong = F.l1_loss(text_features_strong, fixed_embeddings.cuda(), reduction='mean') * lambda2
loss_scl_text_weak = F.l1_loss(text_features_weak, fixed_embeddings.cuda(), reduction='mean') * lambda3 L_sg_weak = F.l1_loss(text_features_weak, fixed_embeddings.cuda(), reduction='mean') * lambda3
L_SCL_logits = F.kl_div( L_zpg = F.kl_div(
F.log_softmax(logits_final / 1, dim=1), F.log_softmax(logits_final / 1, dim=1),
F.log_softmax(zero_shot_logits / 1, dim=1), F.log_softmax(zero_shot_logits / 1, dim=1),
reduction='sum', reduction='sum',
log_target=True log_target=True
) * (1 * 1) / logits_final.numel() ) * (1 * 1) / logits_final.numel()
L_SCL = (L_SCL_logits + loss_scl_text_strong + loss_scl_text_weak + loss_scl_image) L_zg = (L_zpg + L_sg_strong + L_sg_weak + L_zvg)
loss = (loss_ce + L_SCL) loss = (loss_ce + L_zg)
optim.zero_grad() optim.zero_grad()
loss.backward() loss.backward()
optim.step() optim.step()