Fix PathManager Intialization

Author: lolipopshock

src/layoutparser/models/catalog.py

@@ -1,4 +1,6 @@
 from iopath.common.file_io import PathHandler, PathManager, HTTPURLHandler
+from iopath.common.file_io import PathManager as PathManagerBase
+# A trick learned from https://github.com/facebookresearch/detectron2/blob/65faeb4779e4c142484deeece18dc958c5c9ad18/detectron2/utils/file_io.py#L3
 
 MODEL_CATALOG = {
     'HJDataset': {
@@ -61,7 +63,7 @@ class LayoutParserHandler(PathHandler):
     def _get_supported_prefixes(self):
         return [self.PREFIX]
 
-    def _get_local_path(self, path):
+    def _get_local_path(self, path, **kwargs):
         model_name = path[len(self.PREFIX):]
         dataset_name, *model_name, data_type = model_name.split('/')
 
@@ -71,12 +73,12 @@ def _get_local_path(self, path):
             model_url = CONFIG_CATALOG[dataset_name]['/'.join(model_name)]
         else:
             raise ValueError(f"Unknown data_type {data_type}")
-        return PathManager.get_local_path(model_url)
+        return PathManager.get_local_path(model_url, **kwargs)
 
     def _open(self, path, mode="r", **kwargs):
         return PathManager.open(self._get_local_path(path), mode, **kwargs)
 
 
-PathManagerSingleton = PathManager()
-PathManagerSingleton.register_handler(DropboxHandler())
-PathManagerSingleton.register_handler(LayoutParserHandler())
+PathManager = PathManagerBase()
+PathManager.register_handler(DropboxHandler())
+PathManager.register_handler(LayoutParserHandler())

src/layoutparser/models/layoutmodel.py

@@ -6,7 +6,7 @@
 import numpy as np
 import torch
 
-from .catalog import PathManagerSingleton
+from .catalog import PathManager
 from ..elements import *
 
 __all__ = ["Detectron2LayoutModel"]
@@ -93,7 +93,7 @@ class Detectron2LayoutModel(BaseLayoutModel):
     def __init__(self, config_path, model_path=None, label_map=None, extra_config=[]):
 
         cfg = self._config.get_cfg()
-        config_path = PathManagerSingleton.get_local_path(config_path)
+        config_path = PathManager.get_local_path(config_path)
         cfg.merge_from_file(config_path)
         cfg.merge_from_list(extra_config)