From 99b124304fa9c8e48cf428db885c964007c1eb57 Mon Sep 17 00:00:00 2001 From: "zuchen.wang" Date: Fri, 29 Oct 2021 11:36:55 +0800 Subject: [PATCH 1/2] change input size to (0,0) --- fastreid/config/defaults.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/fastreid/config/defaults.py b/fastreid/config/defaults.py index 520abbb..63b8732 100644 --- a/fastreid/config/defaults.py +++ b/fastreid/config/defaults.py @@ -149,9 +149,9 @@ _C.KD.EMA.MOMENTUM = 0.999 # ----------------------------------------------------------------------------- _C.INPUT = CN() # Size of the image during training -_C.INPUT.SIZE_TRAIN = [256, 128] +_C.INPUT.SIZE_TRAIN = [0, 0] # Size of the image during test -_C.INPUT.SIZE_TEST = [256, 128] +_C.INPUT.SIZE_TEST = [0, 0] # `True` if cropping is used for data augmentation during training _C.INPUT.CROP = CN({"ENABLED": False}) From c90792042a5c50a08966cc2b3ed221c38f949f0b Mon Sep 17 00:00:00 2001 From: "zuchen.wang" Date: Fri, 29 Oct 2021 11:45:24 +0800 Subject: [PATCH 2/2] make common base config: input, dataloader, solver, test period and batch --- projects/FastShoe/configs/base-pair.yaml | 85 --------------------- projects/FastShoe/configs/base.yaml | 63 +++++++++++++++ projects/FastShoe/configs/r101-se-ibn.yaml | 89 ---------------------- 3 files changed, 63 insertions(+), 174 deletions(-) delete mode 100644 projects/FastShoe/configs/base-pair.yaml create mode 100644 projects/FastShoe/configs/base.yaml delete mode 100644 projects/FastShoe/configs/r101-se-ibn.yaml diff --git a/projects/FastShoe/configs/base-pair.yaml b/projects/FastShoe/configs/base-pair.yaml deleted file mode 100644 index dd09fbf..0000000 --- a/projects/FastShoe/configs/base-pair.yaml +++ /dev/null @@ -1,85 +0,0 @@ -MODEL: - META_ARCHITECTURE: Metric - - BACKBONE: - NAME: build_resnet_backbone - DEPTH: 101x - NORM: BN - LAST_STRIDE: 2 - FEAT_DIM: 512 - PRETRAIN: True - WITH_IBN: True - WITH_SE: True - - HEADS: - NAME: PairHead - WITH_BNNECK: False - EMBEDDING_DIM: 0 - POOL_LAYER: FastGlobalAvgPool - CLS_LAYER: Linear - NUM_CLASSES: 2 - - LOSSES: - NAME: ("ContrastiveLoss",) - - CONTRASTIVE: - MARGIN: 2.0 - SCALE: 1. - -INPUT: - SIZE_TRAIN: [0,] # no need for resize when training - SIZE_TEST: [256,] - - AUTOAUG: - ENABLED: True - PROB: 0.5 - - CJ: - ENABLED: True - - CROP: - ENABLED: False - SIZE: [224,] - SCALE: [0.08, 1] - RATIO: [0.75, 1.333333333] - - FLIP: - ENABLED: True - -DATALOADER: - SAMPLER_TRAIN: TrainingSampler - NUM_WORKERS: 8 - -SOLVER: - MAX_EPOCH: 1000 - AMP: - ENABLED: True - - OPT: SGD - SCHED: CosineAnnealingLR - - BASE_LR: 0.001 - MOMENTUM: 0.9 - NESTEROV: False - - BIAS_LR_FACTOR: 1. - WEIGHT_DECAY: 0.0005 - WEIGHT_DECAY_BIAS: 0. - IMS_PER_BATCH: 16 - - ETA_MIN_LR: 0.00003 - - WARMUP_FACTOR: 0.1 - WARMUP_ITERS: 100 - - CHECKPOINT_PERIOD: 1 - -TEST: - EVAL_PERIOD: 1 - IMS_PER_BATCH: 32 - -DATASETS: - NAMES: ("ShoeDataset", "OnlineDataset") - TESTS: ("ShoeDataset", "OnlineDataset") - -OUTPUT_DIR: projects/FastShoe/logs/r101_ibn_se diff --git a/projects/FastShoe/configs/base.yaml b/projects/FastShoe/configs/base.yaml new file mode 100644 index 0000000..f4cca7c --- /dev/null +++ b/projects/FastShoe/configs/base.yaml @@ -0,0 +1,63 @@ +INPUT: + SIZE_TRAIN: [0, 0] # resize size + SIZE_TEST: [0, 0] + + CROP: + ENABLED: False + + FLIP: + ENABLED: False + + PADDING: + ENABLED: False + + CJ: + ENABLED: False + + AFFINE: + ENABLED: False + + AUTOAUG: + ENABLED: False + + AUGMIX: + ENABLED: False + + REA: + ENABLED: False + + RPT: + ENABLED: False + + +DATALOADER: + SAMPLER_TRAIN: TrainingSampler + NUM_WORKERS: 8 + +SOLVER: + MAX_EPOCH: 1000 + AMP: + ENABLED: True + + OPT: SGD + SCHED: CosineAnnealingLR + + BASE_LR: 0.001 + MOMENTUM: 0.9 + NESTEROV: False + + BIAS_LR_FACTOR: 1. + WEIGHT_DECAY: 0.0005 + WEIGHT_DECAY_BIAS: 0. + IMS_PER_BATCH: 16 + + ETA_MIN_LR: 0.00003 + + WARMUP_FACTOR: 0.1 + WARMUP_ITERS: 100 + + CHECKPOINT_PERIOD: 1 + +TEST: + EVAL_PERIOD: 1 + IMS_PER_BATCH: 32 diff --git a/projects/FastShoe/configs/r101-se-ibn.yaml b/projects/FastShoe/configs/r101-se-ibn.yaml deleted file mode 100644 index ca005bf..0000000 --- a/projects/FastShoe/configs/r101-se-ibn.yaml +++ /dev/null @@ -1,89 +0,0 @@ -_BASE_: base-pair.yaml - -MODEL: - META_ARCHITECTURE: Metric - - BACKBONE: - NAME: build_resnet_backbone - DEPTH: 101x - NORM: BN - LAST_STRIDE: 2 - FEAT_DIM: 512 - PRETRAIN: True - WITH_IBN: True - WITH_SE: True - - HEADS: - NAME: PairHead - WITH_BNNECK: False - EMBEDDING_DIM: 0 - POOL_LAYER: FastGlobalAvgPool - CLS_LAYER: Linear - NUM_CLASSES: 2 - - LOSSES: - NAME: ("ContrastiveLoss",) - - CONTRASTIVE: - MARGIN: 2.0 - SCALE: 1. - -INPUT: - SIZE_TRAIN: [224, 224] # no need for resize when training - SIZE_TEST: [224, 224] - - FLIP: - ENABLED: True - - CJ: - ENABLED: True - - AUTOAUG: - ENABLED: True - PROB: 0.5 - - AUGMIX: - ENABLED: True - PROB: 0.5 - - REA: - ENABLED: True - - -DATALOADER: - SAMPLER_TRAIN: TrainingSampler - NUM_WORKERS: 8 - -SOLVER: - MAX_EPOCH: 1000 - AMP: - ENABLED: True - - OPT: SGD - SCHED: CosineAnnealingLR - - BASE_LR: 0.001 - MOMENTUM: 0.9 - NESTEROV: False - - BIAS_LR_FACTOR: 1. - WEIGHT_DECAY: 0.0005 - WEIGHT_DECAY_BIAS: 0. - IMS_PER_BATCH: 16 - - ETA_MIN_LR: 0.00003 - - WARMUP_FACTOR: 0.1 - WARMUP_ITERS: 100 - - CHECKPOINT_PERIOD: 1 - -TEST: - EVAL_PERIOD: 1 - IMS_PER_BATCH: 32 - -DATASETS: - NAMES: ("ShoeDataset",) - TESTS: ("ShoeDataset", "OnlineDataset") - -OUTPUT_DIR: projects/FastShoe/logs/r101_ibn_se