30 lines
887 B
YAML
Executable File
30 lines
887 B
YAML
Executable File
MODEL:
|
|
NUM_CLASSES: 21
|
|
BACKBONE:
|
|
OUT_CHANNELS: (512, 1024, 512, 256, 256, 256, 256)
|
|
PRIORS:
|
|
FEATURE_MAPS: [64, 32, 16, 8, 4, 2, 1]
|
|
STRIDES: [8, 16, 32, 64, 128, 256, 512]
|
|
MIN_SIZES: [35.84, 76.8, 153.6, 230.4, 307.2, 384.0, 460.8]
|
|
MAX_SIZES: [76.8, 153.6, 230.4, 307.2, 384.0, 460.8, 537.65]
|
|
ASPECT_RATIOS: [[2], [2, 3], [2, 3], [2, 3], [2, 3], [2], [2]]
|
|
BOXES_PER_LOCATION: [4, 6, 6, 6, 6, 4, 4]
|
|
INPUT:
|
|
IMAGE_SIZE: 512
|
|
DATASETS:
|
|
#TRAIN: ("voc_2007_birdtrain", "voc_2007_birdval")
|
|
#TRAIN: ("voc_2007_trainval", )
|
|
#TEST: ("voc_2007_test", )
|
|
#TEST: ("vocbird_2019_test", )
|
|
TRAIN: ("voc_3_train",)
|
|
TEST: ("voc_3_test", )
|
|
INF: ("inference_hatcher",)
|
|
SOLVER:
|
|
MAX_ITER: 175000
|
|
|
|
LR_STEPS: [80000, 100000]
|
|
GAMMA: 0.1
|
|
BATCH_SIZE: 16
|
|
LR: 1e-5
|
|
#1e-3
|
|
OUTPUT_DIR: '/data1/Hatcher/model/data_MPKFNet/weigth/detection/vgg_ssd512_voc0712VOC3/' |