support open-mmlab#6915: fix format problem

mmdet/utils/misc.py

@@ -1,12 +1,13 @@
 # Copyright (c) OpenMMLab. All rights reserved.
-import os
 import glob
+import os
 import os.path as osp
 import warnings
 
 import mmcv
 from mmcv.utils import print_log
 
+
 def find_latest_checkpoint(path, suffix='pth'):
     """Find the latest checkpoint from the working directory.
 
@@ -42,31 +43,29 @@ def find_latest_checkpoint(path, suffix='pth'):
 
 
 def update_data_root(cfg, logger=None):
-    """Update data root according to environmental variable MMDET_DATASETS.
+    """Update data root according to env MMDET_DATASETS.
 
-    If set env MMDET_DATASETS, update cfg.data_root according to MMDET_DATASETS.
-    Otherwise, using cfg.data_root as default.
+    If set env MMDET_DATASETS, update cfg.data_root according to
+    MMDET_DATASETS. Otherwise, using cfg.data_root as default.
 
     Args:
         cfg (mmcv.Config): The model config need to modify
-        data_root_n (str): The data root set 
         logger (logging.Logger | str | None): the way to print msg
-
     """
     if 'MMDET_DATASETS' not in os.environ:
         print_log(
-            f"Not set env MMDET_DATASETS, using {cfg.data_root} as data root.",
+            f'Not set env MMDET_DATASETS, using {cfg.data_root} as data root.',
             logger)
         return None
     else:
         dst_root = os.environ['MMDET_DATASETS']
         print_log(
-            f"Set env MMDET_DATASETS={dst_root}, using MMDET_DATASETS as data root.",
-            logger)
+            f'Set env MMDET_DATASETS={dst_root}, \
+                using MMDET_DATASETS as data root.', logger)
 
     assert isinstance(cfg, mmcv.Config), \
-        f"cfg got wrong type: {type(cfg)}, expected mmcv.Config"
-    
+        f'cfg got wrong type: {type(cfg)}, expected mmcv.Config'
+
     def update(cfg, src_str, dst_str):
         for k, v in cfg.items():
             if isinstance(v, mmcv.ConfigDict):
@@ -75,4 +74,4 @@ def update(cfg, src_str, dst_str):
                 cfg[k] = v.replace(src_str, dst_str)
 
     update(cfg.data, cfg.data_root, dst_root)
-    cfg.data_root = dst_root
+    cfg.data_root = dst_root

tools/misc/browse_dataset.py

@@ -11,9 +11,9 @@
 from mmdet.core.utils import mask2ndarray
 from mmdet.core.visualization import imshow_det_bboxes
 from mmdet.datasets.builder import build_dataset
-
 from mmdet.utils import update_data_root
 
+
 def parse_args():
     parser = argparse.ArgumentParser(description='Browse a dataset')
     parser.add_argument('config', help='train config file path')

tools/test.py

@@ -17,9 +17,8 @@
 from mmdet.datasets import (build_dataloader, build_dataset,
                             replace_ImageToTensor)
 from mmdet.models import build_detector
-from mmdet.utils import setup_multi_processes
+from mmdet.utils import setup_multi_processes, update_data_root
 
-from mmdet.utils import update_data_root
 
 def parse_args():
     parser = argparse.ArgumentParser(

tools/train.py

@@ -16,9 +16,9 @@
 from mmdet.apis import init_random_seed, set_random_seed, train_detector
 from mmdet.datasets import build_dataset
 from mmdet.models import build_detector
-from mmdet.utils import collect_env, get_root_logger, setup_multi_processes
+from mmdet.utils import (collect_env, get_root_logger, setup_multi_processes,
+                         update_data_root)
 
-from mmdet.utils import update_data_root
 
 def parse_args():
     parser = argparse.ArgumentParser(description='Train a detector')
@@ -102,7 +102,7 @@ def main():
 
     # update data root according to MMDET_DATASETS
     update_data_root(cfg)
-    
+
     if args.cfg_options is not None:
         cfg.merge_from_dict(args.cfg_options)