Skip to content

Commit 307d0e7

Browse files
committed
add sgg config files
1 parent f67634f commit 307d0e7

File tree

2 files changed

+56
-0
lines changed

2 files changed

+56
-0
lines changed

configs/sgg_res101_joint.yaml

+28
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,28 @@
1+
DATASET:
2+
NAME: "vg"
3+
MODE: "benchmark"
4+
TRAIN_BATCH_SIZE: 8
5+
TEST_BATCH_SIZE: 1
6+
MODEL:
7+
WEIGHT_IMG: "catalog://ImageNetPretrained/MSRA/R-101"
8+
WEIGHT_DET: ""
9+
RELATION_ON: True
10+
ALGORITHM: "sg_baseline"
11+
USE_FREQ_PRIOR: False
12+
BACKBONE:
13+
CONV_BODY: "R-101-C4"
14+
FREEZE_PARAMETER: False
15+
RPN:
16+
FREEZE_PARAMETER: False
17+
ROI_HEADS:
18+
BATCH_SIZE_PER_IMAGE: 512
19+
ROI_BOX_HEAD:
20+
NUM_CLASSES: 151
21+
FREEZE_PARAMETER: False
22+
ROI_RELATION_HEAD:
23+
NUM_CLASSES: 51
24+
SOLVER:
25+
BASE_LR: 5e-3
26+
MAX_ITER: 100000
27+
STEPS: (70000,90000)
28+
CHECKPOINT_PERIOD: 5000

configs/sgg_res101_step.yaml

+28
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,28 @@
1+
DATASET:
2+
NAME: "vg"
3+
MODE: "benchmark"
4+
TRAIN_BATCH_SIZE: 8
5+
TEST_BATCH_SIZE: 1
6+
MODEL:
7+
WEIGHT_IMG: "catalog://ImageNetPretrained/MSRA/R-101"
8+
WEIGHT_DET: "checkpoints/vg_benchmark_object/R-101-C4/faster_rcnn/BatchSize_6/Base_LR_0.005/checkpoint_0099999.pth"
9+
RELATION_ON: True
10+
ALGORITHM: "sg_baseline"
11+
USE_FREQ_PRIOR: False
12+
BACKBONE:
13+
CONV_BODY: "R-101-C4"
14+
FREEZE_PARAMETER: True
15+
RPN:
16+
FREEZE_PARAMETER: True
17+
ROI_HEADS:
18+
BATCH_SIZE_PER_IMAGE: 512
19+
ROI_BOX_HEAD:
20+
NUM_CLASSES: 151
21+
FREEZE_PARAMETER: True
22+
ROI_RELATION_HEAD:
23+
NUM_CLASSES: 51
24+
SOLVER:
25+
BASE_LR: 5e-3
26+
MAX_ITER: 40000
27+
STEPS: (20000,30000)
28+
CHECKPOINT_PERIOD: 5000

0 commit comments

Comments
 (0)