[FIx]Replace mmcv.fileclient to mmengine.fileclient (#1737)

* fix mmcv file client

* fix set_env

* fix mmcv file load and dump

tools/dataset_converters/update_infos_to_v2.py

@@ -13,6 +13,7 @@ import time
 from os import path as osp
 
 import mmcv
+import mmengine
 import numpy as np
 from nuscenes.nuscenes import NuScenes
 
@@ -257,7 +258,7 @@ def update_nuscenes_infos(pkl_path, out_dir):
         print(f'Warning, you may overwriting '
               f'the original data {pkl_path}.')
     print(f'Reading from input file: {pkl_path}.')
-    data_list = mmcv.load(pkl_path)
+    data_list = mmengine.load(pkl_path)
     METAINFO = {
         'CLASSES':
         ('car', 'truck', 'trailer', 'bus', 'construction_vehicle', 'bicycle',
@@ -365,7 +366,7 @@ def update_nuscenes_infos(pkl_path, out_dir):
     print(f'ignore classes: {ignore_class_name}')
     converted_data_info = dict(metainfo=METAINFO, data_list=converted_list)
 
-    mmcv.dump(converted_data_info, out_path, 'pkl')
+    mmengine.dump(converted_data_info, out_path, 'pkl')
 
 
 def update_kitti_infos(pkl_path, out_dir):
@@ -381,7 +382,7 @@ def update_kitti_infos(pkl_path, out_dir):
                     'Person_sitting', 'Tram', 'Misc'),
     }
     print(f'Reading from input file: {pkl_path}.')
-    data_list = mmcv.load(pkl_path)
+    data_list = mmengine.load(pkl_path)
     print('Start updating:')
     converted_list = []
     for ori_info_dict in mmcv.track_iter_progress(data_list):
@@ -494,7 +495,7 @@ def update_kitti_infos(pkl_path, out_dir):
     converted_data_info = dict(
         metainfo={'DATASET': 'KITTI'}, data_list=converted_list)
 
-    mmcv.dump(converted_data_info, out_path, 'pkl')
+    mmengine.dump(converted_data_info, out_path, 'pkl')
 
 
 def update_s3dis_infos(pkl_path, out_dir):
@@ -505,7 +506,7 @@ def update_s3dis_infos(pkl_path, out_dir):
         time.sleep(5)
     METAINFO = {'CLASSES': ('table', 'chair', 'sofa', 'bookcase', 'board')}
     print(f'Reading from input file: {pkl_path}.')
-    data_list = mmcv.load(pkl_path)
+    data_list = mmengine.load(pkl_path)
     print('Start updating:')
     converted_list = []
     for i, ori_info_dict in enumerate(mmcv.track_iter_progress(data_list)):
@@ -555,7 +556,7 @@ def update_s3dis_infos(pkl_path, out_dir):
     converted_data_info = dict(
         metainfo={'DATASET': 'S3DIS'}, data_list=converted_list)
 
-    mmcv.dump(converted_data_info, out_path, 'pkl')
+    mmengine.dump(converted_data_info, out_path, 'pkl')
 
 
 def update_scannet_infos(pkl_path, out_dir):
@@ -571,7 +572,7 @@ def update_scannet_infos(pkl_path, out_dir):
          'showercurtrain', 'toilet', 'sink', 'bathtub', 'garbagebin')
     }
     print(f'Reading from input file: {pkl_path}.')
-    data_list = mmcv.load(pkl_path)
+    data_list = mmengine.load(pkl_path)
     print('Start updating:')
     converted_list = []
     for ori_info_dict in mmcv.track_iter_progress(data_list):
@@ -620,7 +621,7 @@ def update_scannet_infos(pkl_path, out_dir):
     converted_data_info = dict(
         metainfo={'DATASET': 'SCANNET'}, data_list=converted_list)
 
-    mmcv.dump(converted_data_info, out_path, 'pkl')
+    mmengine.dump(converted_data_info, out_path, 'pkl')
 
 
 def update_sunrgbd_infos(pkl_path, out_dir):
@@ -634,7 +635,7 @@ def update_sunrgbd_infos(pkl_path, out_dir):
                     'dresser', 'night_stand', 'bookshelf', 'bathtub')
     }
     print(f'Reading from input file: {pkl_path}.')
-    data_list = mmcv.load(pkl_path)
+    data_list = mmengine.load(pkl_path)
     print('Start updating:')
     converted_list = []
     for ori_info_dict in mmcv.track_iter_progress(data_list):
@@ -689,7 +690,7 @@ def update_sunrgbd_infos(pkl_path, out_dir):
     converted_data_info = dict(
         metainfo={'DATASET': 'SUNRGBD'}, data_list=converted_list)
 
-    mmcv.dump(converted_data_info, out_path, 'pkl')
+    mmengine.dump(converted_data_info, out_path, 'pkl')
 
 
 def update_lyft_infos(pkl_path, out_dir):
@@ -698,7 +699,7 @@ def update_lyft_infos(pkl_path, out_dir):
         print(f'Warning, you may overwriting '
               f'the original data {pkl_path}.')
     print(f'Reading from input file: {pkl_path}.')
-    data_list = mmcv.load(pkl_path)
+    data_list = mmengine.load(pkl_path)
     METAINFO = {
         'CLASSES':
         ('car', 'truck', 'bus', 'emergency_vehicle', 'other_vehicle',
@@ -793,7 +794,7 @@ def update_lyft_infos(pkl_path, out_dir):
     print(f'ignore classes: {ignore_class_name}')
     converted_data_info = dict(metainfo=METAINFO, data_list=converted_list)
 
-    mmcv.dump(converted_data_info, out_path, 'pkl')
+    mmengine.dump(converted_data_info, out_path, 'pkl')
 
 
 def update_waymo_infos(pkl_path, out_dir):
@@ -817,7 +818,7 @@ def update_waymo_infos(pkl_path, out_dir):
         'CLASSES': ('Car', 'Pedestrian', 'Cyclist', 'Sign'),
     }
     print(f'Reading from input file: {pkl_path}.')
-    data_list = mmcv.load(pkl_path)
+    data_list = mmengine.load(pkl_path)
     print('Start updating:')
     converted_list = []
     for ori_info_dict in mmcv.track_iter_progress(data_list):
@@ -992,7 +993,7 @@ def update_waymo_infos(pkl_path, out_dir):
     converted_data_info = dict(
         metainfo={'DATASET': 'Waymo'}, data_list=converted_list)
 
-    mmcv.dump(converted_data_info, out_path, 'pkl')
+    mmengine.dump(converted_data_info, out_path, 'pkl')
 
 
 def generate_waymo_camera_instances(ori_info_dict, cam_keys):

tools/misc/visualize_results.py

 # Copyright (c) OpenMMLab. All rights reserved.
 import argparse
 
-import mmcv
+import mmengine
 from mmcv import Config
 
 from mmdet3d.datasets import build_dataset
@@ -31,7 +31,7 @@ def main():
 
     # build the dataset
     dataset = build_dataset(cfg.data.test)
-    results = mmcv.load(args.result)
+    results = mmengine.load(args.result)
 
     if getattr(dataset, 'show', None) is not None:
         # data loading pipeline for showing

tools/update_data_coords.py

@@ -3,6 +3,7 @@ import time
 from os import path as osp
 
 import mmcv
+import mmengine
 import numpy as np
 
 from mmdet3d.structures import limit_period
@@ -18,7 +19,7 @@ def update_sunrgbd_infos(root_dir, out_dir, pkl_files):
     for pkl_file in pkl_files:
         in_path = osp.join(root_dir, pkl_file)
         print(f'Reading from input file: {in_path}.')
-        a = mmcv.load(in_path)
+        a = mmengine.load(in_path)
         print('Start updating:')
         for item in mmcv.track_iter_progress(a):
             if 'rotation_y' in item['annos']:
@@ -28,7 +29,7 @@ def update_sunrgbd_infos(root_dir, out_dir, pkl_files):
 
         out_path = osp.join(out_dir, pkl_file)
         print(f'Writing to output file: {out_path}.')
-        mmcv.dump(a, out_path, 'pkl')
+        mmengine.dump(a, out_path, 'pkl')
 
 
 def update_outdoor_dbinfos(root_dir, out_dir, pkl_files):
@@ -41,7 +42,7 @@ def update_outdoor_dbinfos(root_dir, out_dir, pkl_files):
     for pkl_file in pkl_files:
         in_path = osp.join(root_dir, pkl_file)
         print(f'Reading from input file: {in_path}.')
-        a = mmcv.load(in_path)
+        a = mmengine.load(in_path)
         print('Start updating:')
         for k in a.keys():
             print(f'Updating samples of class {k}:')
@@ -57,7 +58,7 @@ def update_outdoor_dbinfos(root_dir, out_dir, pkl_files):
 
         out_path = osp.join(out_dir, pkl_file)
         print(f'Writing to output file: {out_path}.')
-        mmcv.dump(a, out_path, 'pkl')
+        mmengine.dump(a, out_path, 'pkl')
 
 
 def update_nuscenes_or_lyft_infos(root_dir, out_dir, pkl_files):
@@ -71,7 +72,7 @@ def update_nuscenes_or_lyft_infos(root_dir, out_dir, pkl_files):
     for pkl_file in pkl_files:
         in_path = osp.join(root_dir, pkl_file)
         print(f'Reading from input file: {in_path}.')
-        a = mmcv.load(in_path)
+        a = mmengine.load(in_path)
         print('Start updating:')
         for item in mmcv.track_iter_progress(a['infos']):
             boxes = item['gt_boxes'].copy()
@@ -85,7 +86,7 @@ def update_nuscenes_or_lyft_infos(root_dir, out_dir, pkl_files):
 
         out_path = osp.join(out_dir, pkl_file)
         print(f'Writing to output file: {out_path}.')
-        mmcv.dump(a, out_path, 'pkl')
+        mmengine.dump(a, out_path, 'pkl')
 
 
 parser = argparse.ArgumentParser(description='Arg parser for data coords '