crosstyan/OpenGait
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Update BigGait_CCPG.yaml

Browse Source
This commit is contained in:
Dongyang Jin
2024-05-19 21:53:42 +08:00
committed by GitHub
parent 0e7676a5c2
commit 3d643accfa
1 changed files with 4 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files
configs/biggait/BigGait_CCPG.yaml
+4 -4
View File
@@ -1,8 +1,8 @@
data_cfg: data_cfg:
dataset_name: CCPG dataset_name: CCPG
# TODO # TODO
dataset_root: /4GPU/data/CCPG/Released/CCPG-ratio-pkl/ dataset_root: your_path
dataset_partition: datasets/CCPG/CCPG.json dataset_partition: ./datasets/CCPG/CCPG.json
data_in_use: [True, True] # images / real_ratios data_in_use: [True, True] # images / real_ratios
num_workers: 8 num_workers: 8
remove_no_gallery: false # Remove probe if no gallery for it remove_no_gallery: false # Remove probe if no gallery for it
@@ -37,9 +37,9 @@ loss_cfg:
model_cfg: model_cfg:
model: BigGait__Dinov2_Gaitbase model: BigGait__Dinov2_Gaitbase
pretrained_dinov2: pretrained_LVMs/dinov2_vits14_pretrain.pth # DINOv2 Download Link: https://dl.fbaipublicfiles.com/dinov2/dinov2_vits14/dinov2_vits14_pretrain.pth pretrained_dinov2: ./pretrained_LVMs/dinov2_vits14_pretrain.pth # DINOv2 Download Link: https://dl.fbaipublicfiles.com/dinov2/dinov2_vits14/dinov2_vits14_pretrain.pth
pretrained_mask_branch: None pretrained_mask_branch: None
# pretrained_mask_branch: pretrained_LVMs/MaskBranch_vits14.pt # load pretrain Mask_Branch # pretrained_mask_branch: ./pretrained_LVMs/MaskBranch_vits14.pt # load pretrain Mask_Branch
image_size: 224 # 448x224 image_size: 224 # 448x224
sils_size: 32 # 64x32 sils_size: 32 # 64x32