mirrors/fast-reid
1
0
Fork 0
mirror of https://github.com/JDAI-CV/fast-reid.git synced 2025-06-03 14:50:47 +08:00
Code Issues Projects Releases Wiki Activity

change config file

Browse Source
This commit is contained in:
liaoxingyu 2020-09-10 10:58:56 +08:00
parent 4d573b8107
commit 3c48bf78c1
3 changed files with 6 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
configs/Base-MGN.yml
View File

@ -9,9 +9,7 @@ MODEL:
WITH_NL: False WITH_NL: False
HEADS: HEADS:
NAME: "ReductionHead" EMBEDDING_DIM: 256
POOL_LAYER: "maxpool"
REDUCTION_DIM: 256
LOSSES: LOSSES:
NAME: ("CrossEntropyLoss", "TripletLoss",) NAME: ("CrossEntropyLoss", "TripletLoss",)

2
configs/Base-Strongerbaseline.yml
View File

@ -8,7 +8,7 @@ MODEL:
HEADS: HEADS:
NECK_FEAT: "after" NECK_FEAT: "after"
POOL_LAYER: "gempool" POOL_LAYER: "gempoolP"
CLS_LAYER: "circleSoftmax" CLS_LAYER: "circleSoftmax"
SCALE: 64 SCALE: 64
MARGIN: 0.35 MARGIN: 0.35

5
configs/Base-bagtricks.yml
View File

@ -6,12 +6,15 @@ MODEL:
NORM: "BN" NORM: "BN"
DEPTH: "50x" DEPTH: "50x"
LAST_STRIDE: 1 LAST_STRIDE: 1
FEAT_DIM: 2048
WITH_IBN: False WITH_IBN: False
PRETRAIN: True PRETRAIN: True
PRETRAIN_PATH: "/export/home/lxy/.cache/torch/checkpoints/resnet50-19c8e357.pth"
HEADS: HEADS:
NAME: "BNneckHead" NAME: "EmbeddingHead"
NORM: "BN" NORM: "BN"
WITH_BNNECK: True
POOL_LAYER: "avgpool" POOL_LAYER: "avgpool"
NECK_FEAT: "before" NECK_FEAT: "before"
CLS_LAYER: "linear" CLS_LAYER: "linear"