diff --git a/config.json b/config.json
index 4520898..1b65168 100644
--- a/config.json
+++ b/config.json
@@ -13,7 +13,10 @@
     ],
     "MODEL.BACKBONE.NAME": "build_resnet_fpn_backbone",
     "MODEL.FPN.IN_FEATURES": [
-      "res2"
+      "res2",
+      "res3",
+      "res4",
+      "res5"
     ],
     "MODEL.MASK_ON": true,
     "MODEL.PIXEL_STD": [
@@ -32,11 +35,26 @@
     "MODEL.RESNETS.DEPTH": 101,
     "MODEL.RESNETS.NUM_GROUPS": 32,
     "MODEL.RESNETS.OUT_FEATURES": [
-      "res2"
+      "res2",
+      "res3",
+      "res4",
+      "res5"
     ],
     "MODEL.RESNETS.SIZES": [
       [
         32
+      ],
+      [
+        64
+      ],
+      [
+        128
+      ],
+      [
+        256
+      ],
+      [
+        512
       ]
     ],
     "MODEL.RESNETS.STRIDE_IN_1X1": false,
@@ -45,7 +63,10 @@
     "MODEL.ROI_BOX_HEAD.NUM_FC": 2,
     "MODEL.ROI_BOX_HEAD.POOLER_RESOLUTION": 14,
     "MODEL.ROI_HEADS.IN_FEATURES": [
-      "p2"
+      "p2",
+      "p3",
+      "p4",
+      "p5"
     ],
     "MODEL.ROI_HEADS.NAME": "StandardROIHeads",
     "MODEL.ROI_HEADS.NUM_CLASSES": 5,
@@ -54,7 +75,10 @@
     "MODEL.ROI_MASK_HEAD.POOLER_RESOLUTION": 7,
     "MODEL.RPN.IN_FEATURES": [
       "p2",
-      "p3"
+      "p3",
+      "p4",
+      "p5",
+      "p6"
     ],
     "MODEL.RPN.POST_NMS_TOPK_TRAIN": 1000,
     "MODEL.RPN.PRE_NMS_TOPK_TEST": 1000,
@@ -68,8 +92,8 @@
   "hidden_dropout_prob": 0.1,
   "hidden_size": 96,
   "image_feature_pool_shape": [
-    2,
-    2,
+    7,
+    7,
     256
   ],
   "initializer_range": 0.02,