| | |
| | from iopath.common.file_io import HTTPURLHandler, OneDrivePathHandler, PathHandler |
| | from iopath.common.file_io import PathManager as PathManagerBase |
| |
|
| | __all__ = ["PathManager", "PathHandler"] |
| |
|
| |
|
| | PathManager = PathManagerBase() |
| | """ |
| | This is a detectron2 project-specific PathManager. |
| | We try to stay away from global PathManager in fvcore as it |
| | introduces potential conflicts among other libraries. |
| | """ |
| |
|
| |
|
| | class Detectron2Handler(PathHandler): |
| | """ |
| | Resolve anything that's hosted under detectron2's namespace. |
| | """ |
| |
|
| | PREFIX = "detectron2://" |
| | S3_DETECTRON2_PREFIX = "https://dl.fbaipublicfiles.com/detectron2/" |
| |
|
| | def _get_supported_prefixes(self): |
| | return [self.PREFIX] |
| |
|
| | def _get_local_path(self, path, **kwargs): |
| | name = path[len(self.PREFIX) :] |
| | return PathManager.get_local_path(self.S3_DETECTRON2_PREFIX + name, **kwargs) |
| |
|
| | def _open(self, path, mode="r", **kwargs): |
| | return PathManager.open( |
| | self.S3_DETECTRON2_PREFIX + path[len(self.PREFIX) :], mode, **kwargs |
| | ) |
| |
|
| |
|
| | PathManager.register_handler(HTTPURLHandler()) |
| | PathManager.register_handler(OneDrivePathHandler()) |
| | PathManager.register_handler(Detectron2Handler()) |
| |
|
