migrated to iopath for d2go - data.

Summary: migrated to iopath for d2go - data.

Reviewed By: sstsai-adl

Differential Revision: D26632143

fbshipit-source-id: d85e4a00817115a8762e4444406d867070d4fd0c

d2go/data/cache_util.py

 #!/usr/bin/env python3
 # Copyright (c) Facebook, Inc. and its affiliates. All Rights Reserved
+
 import os
 
+
 def _cache_json_file(json_file):
     # TODO: entirely rely on PathManager for caching
     json_file = os.fspath(json_file)

d2go/data/extended_coco.py

@@ -14,6 +14,7 @@ from detectron2.structures import BoxMode
 from pycocotools.coco import COCO
 
 from .cache_util import _cache_json_file
+
 try:
     # virtual_fs is used to support both local and manifold paths
     # with syntax that is identical to the default python APIs
@@ -272,8 +273,8 @@ def extended_coco_load(json_file, image_root, dataset_name=None, loaded_json=Non
     if dataset_name is not None:
         # overwrite attrs
         meta_dict = MetadataCatalog.get(dataset_name).as_dict()
-        meta_dict['thing_classes'] = classes_to_use
-        meta_dict['thing_dataset_id_to_contiguous_id'] = id_map
+        meta_dict["thing_classes"] = classes_to_use
+        meta_dict["thing_dataset_id_to_contiguous_id"] = id_map
         # update MetadataCatalog (cannot change inplace, has to remove)
         MetadataCatalog.remove(dataset_name)
         MetadataCatalog.get(dataset_name).set(**meta_dict)
@@ -298,10 +299,11 @@ if __name__ == "__main__":
         python -m detectron2.data.datasets.coco \
             path/to/json path/to/image_root dataset_name
     """
+    import sys
+
+    import cv2
     from detectron2.utils.logger import setup_logger
     from detectron2.utils.visualizer import Visualizer
-    import cv2
-    import sys
 
     logger = setup_logger(name=__name__)
     meta = MetadataCatalog.get(sys.argv[3])

d2go/data/gans.py

@@ -16,7 +16,7 @@ raw data with fields such as:
 import os
 import json
 import logging
-from fvcore.common.file_io import PathManager
+from detectron2.utils.file_io import PathManager
 from detectron2.data import DatasetCatalog, MetadataCatalog
 
 

d2go/data/utils.py

@@ -20,7 +20,7 @@ logger = logging.getLogger(__name__)
 from d2go.config import temp_defrost
 from d2go.data.datasets import register_dataset_split, ANN_FN, IM_DIR
 from detectron2.data import DatasetCatalog, MetadataCatalog
-from fvcore.common.file_io import PathManager
+from detectron2.utils.file_io import PathManager
 
 
 class AdhocDatasetManager:

d2go/export/api.py

@@ -33,7 +33,7 @@ import torch
 import torch.nn as nn
 import torch.quantization.quantize_fx
 from d2go.modeling.quantization import post_training_quantize
-from fvcore.common.file_io import PathManager
+from detectron2.utils.file_io import PathManager
 from mobile_cv.arch.utils import fuse_utils
 from mobile_cv.predictor.api import FuncInfo, ModelInfo, PredictorInfo
 from mobile_cv.predictor.builtin_functions import (

d2go/export/torchscript.py

@@ -7,7 +7,7 @@ import os
 from typing import Tuple, Optional, Dict
 
 import torch
-from fvcore.common.file_io import PathManager
+from detectron2.utils.file_io import PathManager
 from torch import nn
 
 

d2go/setup.py

@@ -21,8 +21,8 @@ from d2go.utils.launch_environment import get_launch_environment
 from detectron2.utils.collect_env import collect_env_info
 from detectron2.utils.logger import setup_logger
 from detectron2.utils.serialize import PicklableWrapper
-from fvcore.common.file_io import PathManager
 from d2go.utils.helper import run_once
+from detectron2.utils.file_io import PathManager
 from mobile_cv.common.misc.py import FolderLock, MultiprocessingPdb, post_mortem_if_fail
 
 

d2go/utils/abnormal_checker.py

@@ -8,7 +8,7 @@ import os
 import detectron2.utils.comm as comm
 import torch
 from d2go.utils.visualization import VisualizerWrapper
-from fvcore.common.file_io import PathManager
+from detectron2.utils.file_io import PathManager
 
 
 logger = logging.getLogger(__name__)

d2go/utils/validation_monitor.py

@@ -8,7 +8,7 @@ import re
 import time
 
 from detectron2.checkpoint import DetectionCheckpointer
-from fvcore.common.file_io import PathManager
+from detectron2.utils.file_io import PathManager
 from pytorch_lightning.callbacks.model_checkpoint import ModelCheckpoint