rename imports of point_rend and deeplab

Reviewed By: ytsheng Differential Revision: D23015493 fbshipit-source-id: 246c37a9cd913202e801e4e781adb63c4b74c224
facebookresearch · Aug 13, 2020 · 9fe2eda · 9fe2eda
1 parent a7e07a6
commit 9fe2eda
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 4 deletions.
diff --git a/projects/DeepLab/train_net.py b/projects/DeepLab/train_net.py
@@ -17,8 +17,7 @@
 from detectron2.data import DatasetMapper, MetadataCatalog, build_detection_train_loader
 from detectron2.engine import DefaultTrainer, default_argument_parser, default_setup, launch
 from detectron2.evaluation import CityscapesSemSegEvaluator, DatasetEvaluators, SemSegEvaluator
-
-from deeplab import add_deeplab_config, build_lr_scheduler
+from detectron2.projects.deeplab import add_deeplab_config, build_lr_scheduler
 
 
 def build_sem_seg_train_aug(cfg):

diff --git a/projects/PointRend/train_net.py b/projects/PointRend/train_net.py
@@ -25,8 +25,7 @@
     SemSegEvaluator,
     verify_results,
 )
-
-from point_rend import ColorAugSSDTransform, add_pointrend_config
+from detectron2.projects.point_rend import ColorAugSSDTransform, add_pointrend_config
 
 
 def build_sem_seg_train_aug(cfg):