forked from microsoft/scene_graph_benchmark
-
Notifications
You must be signed in to change notification settings - Fork 1
/
R152FPN_reldn_oi.yaml
83 lines (83 loc) · 2.53 KB
/
R152FPN_reldn_oi.yaml
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
MODEL:
META_ARCHITECTURE: "SceneParser"
WEIGHT: "pretrained_model/sgg_model_zoo/oi_R152_reldn.pth"
USE_FREQ_PRIOR: False
FREQ_PRIOR: "openimages_v5c/vrd/vrd_frequency_prior_include_background_update.npy"
BACKBONE:
CONV_BODY: "R-152-FPN"
RESNETS:
BACKBONE_OUT_CHANNELS: 256
STRIDE_IN_1X1: False
NUM_GROUPS: 32
WIDTH_PER_GROUP: 8
RPN:
USE_FPN: True
ANCHOR_STRIDE: (4, 8, 16, 32, 64)
PRE_NMS_TOP_N_TRAIN: 2000
PRE_NMS_TOP_N_TEST: 1000
FPN_POST_NMS_TOP_N_TRAIN: 1000
FPN_POST_NMS_TOP_N_TEST: 1000
ROI_HEADS:
USE_FPN: True
POSITIVE_FRACTION: 0.5
SCORE_THRESH: 0.05
DETECTIONS_PER_IMG: 100
MIN_DETECTIONS_PER_IMG: 0
ROI_BOX_HEAD:
NUM_CLASSES: 58
POOLER_RESOLUTION: 7
POOLER_SCALES: (0.25, 0.125, 0.0625, 0.03125)
POOLER_SAMPLING_RATIO: 2
FEATURE_EXTRACTOR: "FPN2MLPFeatureExtractor"
PREDICTOR: "FPNPredictor"
ATTRIBUTE_ON: False
RELATION_ON: True
ROI_RELATION_HEAD:
DETECTOR_PRE_CALCULATED: True
DETECTOR_BOX_THRESHOLD: 0.05
FORCE_RELATIONS: False
ALGORITHM: "sg_reldn"
MODE: 'sgdet'
USE_BIAS: False
FILTER_NON_OVERLAP: True
UPDATE_BOX_REG: False
SHARE_CONV_BACKBONE: False
SHARE_BOX_FEATURE_EXTRACTOR: False
SEPERATE_SO_FEATURE_EXTRACTOR: True
NUM_CLASSES: 10
POOLER_RESOLUTION: 7
POOLER_SCALES: (0.25, 0.125, 0.0625, 0.03125)
POOLER_SAMPLING_RATIO: 2
FEATURE_EXTRACTOR: "FPN2MLPRelationFeatureExtractor"
PREDICTOR: "FPNRelationPredictor"
CONTRASTIVE_LOSS:
USE_FLAG: True
TRIPLETS_PER_IMG: 100
POSTPROCESS_METHOD: 'unconstrained'
INPUT:
PIXEL_MEAN: [103.530, 116.280, 123.675]
DATASETS:
TRAIN: ("openimages_v5c/vrd/train.vrd.challenge18.R152detector_pre_calculate.yaml",)
TEST: ("openimages_v5c/vrd/val.vrd.challenge18.R152detector_pre_calculate.yaml",)
LABELMAP_FILE: 'openimages_v5c/vrd/ji_vrd_labelmap.json'
DATALOADER:
SIZE_DIVISIBILITY: 32
NUM_WORKERS: 16
SOLVER:
BASE_LR: 0.005
WEIGHT_DECAY: 0.0001
MAX_ITER: 162000
STEPS: (97000, 129600)
IMS_PER_BATCH: 4
CHECKPOINT_PERIOD: 20000
TEST:
IMS_PER_BATCH: 1
SAVE_PREDICTIONS: True
SAVE_RESULTS_TO_TSV: True
TSV_SAVE_SUBSET: ['rect', 'class', 'conf', 'relations', 'relation_scores']
GATHER_ON_CPU: True
SKIP_PERFORMANCE_EVAL: False
OUTPUT_RELATION_FEATURE: False
OUTPUT_DIR: "./exps/ji_relation_X152FPN_test"
DATA_DIR: "./datasets"
DISTRIBUTED_BACKEND: 'gloo'