Add Deeplab + config files

Summary: Adding Deeplab head to Densepose + config files

Reviewed By: vkhalidov

Differential Revision: D19345446

fbshipit-source-id: 3dbcfb38ad4a598a4f7377909d086acaf4a6972d

Author: MarcSzafraniec

projects/DensePose/configs/Base-DensePose-RCNN-FPN.yaml

@@ -44,4 +44,3 @@ SOLVER:
   WARMUP_FACTOR: 0.1
 INPUT:
   MIN_SIZE_TRAIN: (640, 672, 704, 736, 768, 800)
-VERSION: 2

projects/DensePose/configs/densepose_rcnn_R_101_FPN_DL_s1x.yaml

@@ -0,0 +1,10 @@
+_BASE_: "Base-DensePose-RCNN-FPN.yaml"
+MODEL:
+  WEIGHTS: "detectron2://ImageNetPretrained/MSRA/R-101.pkl"
+  RESNETS:
+    DEPTH: 101
+  ROI_DENSEPOSE_HEAD:
+    NAME: "DensePoseDeepLabHead"
+SOLVER:
+  MAX_ITER: 130000
+  STEPS: (100000, 120000)

projects/DensePose/configs/densepose_rcnn_R_101_FPN_s1x.yaml

@@ -6,4 +6,3 @@ MODEL:
 SOLVER:
   MAX_ITER: 130000
   STEPS: (100000, 120000)
-

projects/DensePose/configs/densepose_rcnn_R_50_FPN_DL_s1x.yaml

@@ -0,0 +1,10 @@
+_BASE_: "Base-DensePose-RCNN-FPN.yaml"
+MODEL:
+  WEIGHTS: "detectron2://ImageNetPretrained/MSRA/R-50.pkl"
+  RESNETS:
+    DEPTH: 50
+  ROI_DENSEPOSE_HEAD:
+    NAME: "DensePoseDeepLabHead"
+SOLVER:
+  MAX_ITER: 130000
+  STEPS: (100000, 120000)

projects/DensePose/configs/densepose_rcnn_R_50_FPN_s1x.yaml

@@ -6,4 +6,3 @@ MODEL:
 SOLVER:
   MAX_ITER: 130000
   STEPS: (100000, 120000)
-

projects/DensePose/configs/quick_schedules/densepose_rcnn_R_50_FPN_DL_instant_test.yaml

@@ -0,0 +1,11 @@
+_BASE_: "../Base-DensePose-RCNN-FPN.yaml"
+MODEL:
+  WEIGHTS: "detectron2://ImageNetPretrained/MSRA/R-50.pkl"
+  ROI_DENSEPOSE_HEAD:
+    NAME: "DensePoseDeepLabHead"
+DATASETS:
+  TRAIN: ("densepose_coco_2014_minival_100",)
+  TEST: ("densepose_coco_2014_minival_100",)
+SOLVER:
+  MAX_ITER: 40
+  STEPS: (30,)

projects/DensePose/densepose/config.py

@@ -33,3 +33,7 @@ def add_densepose_config(cfg):
     _C.MODEL.ROI_DENSEPOSE_HEAD.PART_WEIGHTS = 0.3
     # Loss weights for UV regression.
     _C.MODEL.ROI_DENSEPOSE_HEAD.POINT_REGRESSION_WEIGHTS = 0.1
+    # For DeepLab head
+    _C.MODEL.ROI_DENSEPOSE_HEAD.DEEPLAB = CN()
+    _C.MODEL.ROI_DENSEPOSE_HEAD.DEEPLAB.NORM = "GN"
+    _C.MODEL.ROI_DENSEPOSE_HEAD.DEEPLAB.NONLOCAL_ON = 0