Update config.json to include p5 and a few other layers

#1
by ankrgyl - opened
Files changed (1) hide show
  1. config.json +30 -6
config.json CHANGED
@@ -13,7 +13,10 @@
13
  ],
14
  "MODEL.BACKBONE.NAME": "build_resnet_fpn_backbone",
15
  "MODEL.FPN.IN_FEATURES": [
16
- "res2"
 
 
 
17
  ],
18
  "MODEL.MASK_ON": true,
19
  "MODEL.PIXEL_STD": [
@@ -32,11 +35,26 @@
32
  "MODEL.RESNETS.DEPTH": 101,
33
  "MODEL.RESNETS.NUM_GROUPS": 32,
34
  "MODEL.RESNETS.OUT_FEATURES": [
35
- "res2"
 
 
 
36
  ],
37
  "MODEL.RESNETS.SIZES": [
38
  [
39
  32
 
 
 
 
 
 
 
 
 
 
 
 
40
  ]
41
  ],
42
  "MODEL.RESNETS.STRIDE_IN_1X1": false,
@@ -45,7 +63,10 @@
45
  "MODEL.ROI_BOX_HEAD.NUM_FC": 2,
46
  "MODEL.ROI_BOX_HEAD.POOLER_RESOLUTION": 14,
47
  "MODEL.ROI_HEADS.IN_FEATURES": [
48
- "p2"
 
 
 
49
  ],
50
  "MODEL.ROI_HEADS.NAME": "StandardROIHeads",
51
  "MODEL.ROI_HEADS.NUM_CLASSES": 5,
@@ -54,7 +75,10 @@
54
  "MODEL.ROI_MASK_HEAD.POOLER_RESOLUTION": 7,
55
  "MODEL.RPN.IN_FEATURES": [
56
  "p2",
57
- "p3"
 
 
 
58
  ],
59
  "MODEL.RPN.POST_NMS_TOPK_TRAIN": 1000,
60
  "MODEL.RPN.PRE_NMS_TOPK_TEST": 1000,
@@ -68,8 +92,8 @@
68
  "hidden_dropout_prob": 0.1,
69
  "hidden_size": 96,
70
  "image_feature_pool_shape": [
71
- 2,
72
- 2,
73
  256
74
  ],
75
  "initializer_range": 0.02,
 
13
  ],
14
  "MODEL.BACKBONE.NAME": "build_resnet_fpn_backbone",
15
  "MODEL.FPN.IN_FEATURES": [
16
+ "res2",
17
+ "res3",
18
+ "res4",
19
+ "res5"
20
  ],
21
  "MODEL.MASK_ON": true,
22
  "MODEL.PIXEL_STD": [
 
35
  "MODEL.RESNETS.DEPTH": 101,
36
  "MODEL.RESNETS.NUM_GROUPS": 32,
37
  "MODEL.RESNETS.OUT_FEATURES": [
38
+ "res2",
39
+ "res3",
40
+ "res4",
41
+ "res5"
42
  ],
43
  "MODEL.RESNETS.SIZES": [
44
  [
45
  32
46
+ ],
47
+ [
48
+ 64
49
+ ],
50
+ [
51
+ 128
52
+ ],
53
+ [
54
+ 256
55
+ ],
56
+ [
57
+ 512
58
  ]
59
  ],
60
  "MODEL.RESNETS.STRIDE_IN_1X1": false,
 
63
  "MODEL.ROI_BOX_HEAD.NUM_FC": 2,
64
  "MODEL.ROI_BOX_HEAD.POOLER_RESOLUTION": 14,
65
  "MODEL.ROI_HEADS.IN_FEATURES": [
66
+ "p2",
67
+ "p3",
68
+ "p4",
69
+ "p5"
70
  ],
71
  "MODEL.ROI_HEADS.NAME": "StandardROIHeads",
72
  "MODEL.ROI_HEADS.NUM_CLASSES": 5,
 
75
  "MODEL.ROI_MASK_HEAD.POOLER_RESOLUTION": 7,
76
  "MODEL.RPN.IN_FEATURES": [
77
  "p2",
78
+ "p3",
79
+ "p4",
80
+ "p5",
81
+ "p6"
82
  ],
83
  "MODEL.RPN.POST_NMS_TOPK_TRAIN": 1000,
84
  "MODEL.RPN.PRE_NMS_TOPK_TEST": 1000,
 
92
  "hidden_dropout_prob": 0.1,
93
  "hidden_size": 96,
94
  "image_feature_pool_shape": [
95
+ 7,
96
+ 7,
97
  256
98
  ],
99
  "initializer_range": 0.02,