From 61d87c7617ebf865898f31d730e48e72ff9922fd Mon Sep 17 00:00:00 2001
From: Jianfeng Wang <wangjianfeng@megvii.com>
Date: Tue, 28 Jul 2020 10:30:39 +0800
Subject: [PATCH] fix(detection): avoid loading backbone weight when imported
 from hub (#53)

---
 .../detection/configs/faster_rcnn_res101_coco_2x_800size.py   | 4 +++-
 .../detection/configs/faster_rcnn_res50_coco_1x_800size.py    | 4 +++-
 .../configs/faster_rcnn_res50_coco_1x_800size_syncbn.py       | 4 +++-
 .../detection/configs/faster_rcnn_resx101_coco_2x_800size.py  | 4 +++-
 .../detection/configs/retinanet_res101_coco_2x_800size.py     | 4 +++-
 .../detection/configs/retinanet_res50_coco_1x_800size.py      | 4 +++-
 .../configs/retinanet_res50_coco_1x_800size_syncbn.py         | 4 +++-
 .../detection/configs/retinanet_resx101_coco_2x_800size.py    | 4 +++-
 8 files changed, 24 insertions(+), 8 deletions(-)

diff --git a/official/vision/detection/configs/faster_rcnn_res101_coco_2x_800size.py b/official/vision/detection/configs/faster_rcnn_res101_coco_2x_800size.py
index da6c3fd..a3dfcfd 100644
--- a/official/vision/detection/configs/faster_rcnn_res101_coco_2x_800size.py
+++ b/official/vision/detection/configs/faster_rcnn_res101_coco_2x_800size.py
@@ -33,7 +33,9 @@ def faster_rcnn_res101_coco_2x_800size(batch_size=1, **kwargs):
     `"FPN" <https://arxiv.org/abs/1612.03144>`_
     `"COCO" <https://arxiv.org/abs/1405.0312>`_
     """
-    return models.FasterRCNN(CustomFasterRCNNConfig(), batch_size=batch_size, **kwargs)
+    cfg = CustomFasterRCNNConfig()
+    cfg.backbone_pretrained = False
+    return models.FasterRCNN(cfg, batch_size=batch_size, **kwargs)
 
 
 Net = models.FasterRCNN
diff --git a/official/vision/detection/configs/faster_rcnn_res50_coco_1x_800size.py b/official/vision/detection/configs/faster_rcnn_res50_coco_1x_800size.py
index 6eff2fe..702a939 100644
--- a/official/vision/detection/configs/faster_rcnn_res50_coco_1x_800size.py
+++ b/official/vision/detection/configs/faster_rcnn_res50_coco_1x_800size.py
@@ -22,7 +22,9 @@ def faster_rcnn_res50_coco_1x_800size(batch_size=1, **kwargs):
     `"FPN" <https://arxiv.org/abs/1612.03144>`_
     `"COCO" <https://arxiv.org/abs/1405.0312>`_
     """
-    return models.FasterRCNN(models.FasterRCNNConfig(), batch_size=batch_size, **kwargs)
+    cfg = models.FasterRCNNConfig()
+    cfg.backbone_pretrained = False
+    return models.FasterRCNN(cfg, batch_size=batch_size, **kwargs)
 
 
 Net = models.FasterRCNN
diff --git a/official/vision/detection/configs/faster_rcnn_res50_coco_1x_800size_syncbn.py b/official/vision/detection/configs/faster_rcnn_res50_coco_1x_800size_syncbn.py
index 9527c90..744cc22 100644
--- a/official/vision/detection/configs/faster_rcnn_res50_coco_1x_800size_syncbn.py
+++ b/official/vision/detection/configs/faster_rcnn_res50_coco_1x_800size_syncbn.py
@@ -32,7 +32,9 @@ def faster_rcnn_res50_coco_1x_800size_syncbn(batch_size=1, **kwargs):
     `"COCO" <https://arxiv.org/abs/1405.0312>`_
     `"SyncBN" <https://arxiv.org/abs/1711.07240>`_
     """
-    return models.FasterRCNN(CustomFasterRCNNConfig(), batch_size=batch_size, **kwargs)
+    cfg = CustomFasterRCNNConfig()
+    cfg.backbone_pretrained = False
+    return models.FasterRCNN(cfg, batch_size=batch_size, **kwargs)
 
 
 Net = models.FasterRCNN
diff --git a/official/vision/detection/configs/faster_rcnn_resx101_coco_2x_800size.py b/official/vision/detection/configs/faster_rcnn_resx101_coco_2x_800size.py
index 974d2f0..56798d2 100644
--- a/official/vision/detection/configs/faster_rcnn_resx101_coco_2x_800size.py
+++ b/official/vision/detection/configs/faster_rcnn_resx101_coco_2x_800size.py
@@ -33,7 +33,9 @@ def faster_rcnn_resx101_coco_2x_800size(batch_size=1, **kwargs):
     `"FPN" <https://arxiv.org/abs/1612.03144>`_
     `"COCO" <https://arxiv.org/abs/1405.0312>`_
     """
-    return models.FasterRCNN(CustomFasterRCNNConfig(), batch_size=batch_size, **kwargs)
+    cfg = CustomFasterRCNNConfig()
+    cfg.backbone_pretrained = False
+    return models.FasterRCNN(cfg, batch_size=batch_size, **kwargs)
 
 
 Net = models.FasterRCNN
diff --git a/official/vision/detection/configs/retinanet_res101_coco_2x_800size.py b/official/vision/detection/configs/retinanet_res101_coco_2x_800size.py
index 456f705..51230ac 100644
--- a/official/vision/detection/configs/retinanet_res101_coco_2x_800size.py
+++ b/official/vision/detection/configs/retinanet_res101_coco_2x_800size.py
@@ -33,7 +33,9 @@ def retinanet_res101_coco_2x_800size(batch_size=1, **kwargs):
     `"FPN" <https://arxiv.org/abs/1612.03144>`_
     `"COCO" <https://arxiv.org/abs/1405.0312>`_
     """
-    return models.RetinaNet(CustomRetinaNetConfig(), batch_size=batch_size, **kwargs)
+    cfg = CustomRetinaNetConfig()
+    cfg.backbone_pretrained = False
+    return models.RetinaNet(cfg, batch_size=batch_size, **kwargs)
 
 
 Net = models.RetinaNet
diff --git a/official/vision/detection/configs/retinanet_res50_coco_1x_800size.py b/official/vision/detection/configs/retinanet_res50_coco_1x_800size.py
index 3cb0ba4..df9c693 100644
--- a/official/vision/detection/configs/retinanet_res50_coco_1x_800size.py
+++ b/official/vision/detection/configs/retinanet_res50_coco_1x_800size.py
@@ -22,7 +22,9 @@ def retinanet_res50_coco_1x_800size(batch_size=1, **kwargs):
     `"FPN" <https://arxiv.org/abs/1612.03144>`_
     `"COCO" <https://arxiv.org/abs/1405.0312>`_
     """
-    return models.RetinaNet(models.RetinaNetConfig(), batch_size=batch_size, **kwargs)
+    cfg = models.RetinaNetConfig()
+    cfg.backbone_pretrained = False
+    return models.RetinaNet(cfg, batch_size=batch_size, **kwargs)
 
 
 Net = models.RetinaNet
diff --git a/official/vision/detection/configs/retinanet_res50_coco_1x_800size_syncbn.py b/official/vision/detection/configs/retinanet_res50_coco_1x_800size_syncbn.py
index bfd31d2..2e591db 100644
--- a/official/vision/detection/configs/retinanet_res50_coco_1x_800size_syncbn.py
+++ b/official/vision/detection/configs/retinanet_res50_coco_1x_800size_syncbn.py
@@ -32,7 +32,9 @@ def retinanet_res50_coco_1x_800size_syncbn(batch_size=1, **kwargs):
     `"COCO" <https://arxiv.org/abs/1405.0312>`_
     `"SyncBN" <https://arxiv.org/abs/1711.07240>`_
     """
-    return models.RetinaNet(CustomRetinaNetConfig(), batch_size=batch_size, **kwargs)
+    cfg = CustomRetinaNetConfig()
+    cfg.backbone_pretrained = False
+    return models.RetinaNet(cfg, batch_size=batch_size, **kwargs)
 
 
 Net = models.RetinaNet
diff --git a/official/vision/detection/configs/retinanet_resx101_coco_2x_800size.py b/official/vision/detection/configs/retinanet_resx101_coco_2x_800size.py
index 38c5316..e72e016 100644
--- a/official/vision/detection/configs/retinanet_resx101_coco_2x_800size.py
+++ b/official/vision/detection/configs/retinanet_resx101_coco_2x_800size.py
@@ -33,7 +33,9 @@ def retinanet_resx101_coco_2x_800size(batch_size=1, **kwargs):
     `"FPN" <https://arxiv.org/abs/1612.03144>`_
     `"COCO" <https://arxiv.org/abs/1405.0312>`_
     """
-    return models.RetinaNet(CustomRetinaNetConfig(), batch_size=batch_size, **kwargs)
+    cfg = CustomRetinaNetConfig()
+    cfg.backbone_pretrained = False
+    return models.RetinaNet(cfg, batch_size=batch_size, **kwargs)
 
 
 Net = models.RetinaNet
-- 
GitLab