Unverified Commit 6f1a268e authored by ChaimZhu's avatar ChaimZhu Committed by GitHub
Browse files

[Enhancement] Upgrade isort in pre-commit hook (#1217)

* update isort

* update setuptools

* fix

* pre-commit run --all-files
parent 53488dad
# Copyright (c) OpenMMLab. All rights reserved. # Copyright (c) OpenMMLab. All rights reserved.
import mmcv
import numpy as np
import os import os
from logging import warning from logging import warning
from lyft_dataset_sdk.lyftdataset import LyftDataset as Lyft
from os import path as osp from os import path as osp
import mmcv
import numpy as np
from lyft_dataset_sdk.lyftdataset import LyftDataset as Lyft
from pyquaternion import Quaternion from pyquaternion import Quaternion
from mmdet3d.datasets import LyftDataset from mmdet3d.datasets import LyftDataset
......
# Copyright (c) OpenMMLab. All rights reserved. # Copyright (c) OpenMMLab. All rights reserved.
import argparse import argparse
import numpy as np
import os import os
import numpy as np
def fix_lyft(root_folder='./data/lyft', version='v1.01'): def fix_lyft(root_folder='./data/lyft', version='v1.01'):
# refer to https://www.kaggle.com/c/3d-object-detection-for-autonomous-vehicles/discussion/110000 # noqa # refer to https://www.kaggle.com/c/3d-object-detection-for-autonomous-vehicles/discussion/110000 # noqa
......
# Copyright (c) OpenMMLab. All rights reserved. # Copyright (c) OpenMMLab. All rights reserved.
import argparse import argparse
import base64 import base64
from os import path as osp
import mmcv import mmcv
import numpy as np import numpy as np
from nuimages import NuImages from nuimages import NuImages
from nuimages.utils.utils import mask_decode, name_to_index_mapping from nuimages.utils.utils import mask_decode, name_to_index_mapping
from os import path as osp
nus_categories = ('car', 'truck', 'trailer', 'bus', 'construction_vehicle', nus_categories = ('car', 'truck', 'trailer', 'bus', 'construction_vehicle',
'bicycle', 'motorcycle', 'pedestrian', 'traffic_cone', 'bicycle', 'motorcycle', 'pedestrian', 'traffic_cone',
......
# Copyright (c) OpenMMLab. All rights reserved. # Copyright (c) OpenMMLab. All rights reserved.
import mmcv
import numpy as np
import os import os
from collections import OrderedDict from collections import OrderedDict
from os import path as osp
from typing import List, Tuple, Union
import mmcv
import numpy as np
from nuscenes.nuscenes import NuScenes from nuscenes.nuscenes import NuScenes
from nuscenes.utils.geometry_utils import view_points from nuscenes.utils.geometry_utils import view_points
from os import path as osp
from pyquaternion import Quaternion from pyquaternion import Quaternion
from shapely.geometry import MultiPoint, box from shapely.geometry import MultiPoint, box
from typing import List, Tuple, Union
from mmdet3d.core.bbox import points_cam2img from mmdet3d.core.bbox import points_cam2img
from mmdet3d.datasets import NuScenesDataset from mmdet3d.datasets import NuScenesDataset
......
# Copyright (c) OpenMMLab. All rights reserved. # Copyright (c) OpenMMLab. All rights reserved.
import mmcv
import numpy as np
import os import os
from concurrent import futures as futures from concurrent import futures as futures
from os import path as osp from os import path as osp
import mmcv
import numpy as np
class S3DISData(object): class S3DISData(object):
"""S3DIS data. """S3DIS data.
......
# Copyright (c) OpenMMLab. All rights reserved. # Copyright (c) OpenMMLab. All rights reserved.
import mmcv
import numpy as np
import os import os
from concurrent import futures as futures from concurrent import futures as futures
from os import path as osp from os import path as osp
import mmcv
import numpy as np
class ScanNetData(object): class ScanNetData(object):
"""ScanNet data. """ScanNet data.
......
# Copyright (c) OpenMMLab. All rights reserved. # Copyright (c) OpenMMLab. All rights reserved.
import mmcv
import numpy as np
from concurrent import futures as futures from concurrent import futures as futures
from os import path as osp from os import path as osp
import mmcv
import numpy as np
from scipy import io as sio from scipy import io as sio
......
...@@ -10,11 +10,12 @@ except ImportError: ...@@ -10,11 +10,12 @@ except ImportError:
'Please run "pip install waymo-open-dataset-tf-2-1-0==1.2.0" ' 'Please run "pip install waymo-open-dataset-tf-2-1-0==1.2.0" '
'to install the official devkit first.') 'to install the official devkit first.')
from glob import glob
from os.path import join
import mmcv import mmcv
import numpy as np import numpy as np
import tensorflow as tf import tensorflow as tf
from glob import glob
from os.path import join
from waymo_open_dataset.utils import range_image_utils, transform_utils from waymo_open_dataset.utils import range_image_utils, transform_utils
from waymo_open_dataset.utils.frame_utils import \ from waymo_open_dataset.utils.frame_utils import \
parse_range_image_and_camera_projection parse_range_image_and_camera_projection
......
# Copyright (c) OpenMMLab. All rights reserved. # Copyright (c) OpenMMLab. All rights reserved.
import mmcv
from argparse import ArgumentParser, Namespace from argparse import ArgumentParser, Namespace
from pathlib import Path from pathlib import Path
from tempfile import TemporaryDirectory from tempfile import TemporaryDirectory
import mmcv
try: try:
from model_archiver.model_packaging import package_model from model_archiver.model_packaging import package_model
from model_archiver.model_packaging_utils import ModelExportUtils from model_archiver.model_packaging_utils import ModelExportUtils
......
# Copyright (c) OpenMMLab. All rights reserved. # Copyright (c) OpenMMLab. All rights reserved.
import base64 import base64
import numpy as np
import os import os
import numpy as np
import torch import torch
from ts.torch_handler.base_handler import BaseHandler from ts.torch_handler.base_handler import BaseHandler
......
from argparse import ArgumentParser
import numpy as np import numpy as np
import requests import requests
from argparse import ArgumentParser
from mmdet3d.apis import inference_detector, init_model from mmdet3d.apis import inference_detector, init_model
......
# Copyright (c) OpenMMLab. All rights reserved. # Copyright (c) OpenMMLab. All rights reserved.
import argparse import argparse
import mmcv
import numpy as np
import warnings import warnings
from mmcv import Config, DictAction, mkdir_or_exist
from os import path as osp from os import path as osp
from pathlib import Path from pathlib import Path
import mmcv
import numpy as np
from mmcv import Config, DictAction, mkdir_or_exist
from mmdet3d.core.bbox import (Box3DMode, CameraInstance3DBoxes, Coord3DMode, from mmdet3d.core.bbox import (Box3DMode, CameraInstance3DBoxes, Coord3DMode,
DepthInstance3DBoxes, LiDARInstance3DBoxes) DepthInstance3DBoxes, LiDARInstance3DBoxes)
from mmdet3d.core.visualizer import (show_multi_modality_result, show_result, from mmdet3d.core.visualizer import (show_multi_modality_result, show_result,
......
# Copyright (c) OpenMMLab. All rights reserved. # Copyright (c) OpenMMLab. All rights reserved.
import argparse import argparse
import torch import torch
from mmcv.runner import save_checkpoint from mmcv.runner import save_checkpoint
from torch import nn as nn from torch import nn as nn
......
# Copyright (c) OpenMMLab. All rights reserved. # Copyright (c) OpenMMLab. All rights reserved.
import argparse import argparse
from mmcv import Config, DictAction from mmcv import Config, DictAction
......
# Copyright (c) OpenMMLab. All rights reserved. # Copyright (c) OpenMMLab. All rights reserved.
import argparse import argparse
import mmcv import mmcv
from mmcv import Config from mmcv import Config
......
# Copyright (c) OpenMMLab. All rights reserved. # Copyright (c) OpenMMLab. All rights reserved.
import argparse import argparse
import tempfile import tempfile
import torch import torch
from mmcv import Config from mmcv import Config
from mmcv.runner import load_state_dict from mmcv.runner import load_state_dict
......
# Copyright (c) OpenMMLab. All rights reserved. # Copyright (c) OpenMMLab. All rights reserved.
import argparse import argparse
import tempfile import tempfile
import torch import torch
from mmcv import Config from mmcv import Config
from mmcv.runner import load_state_dict from mmcv.runner import load_state_dict
......
# Copyright (c) OpenMMLab. All rights reserved. # Copyright (c) OpenMMLab. All rights reserved.
import argparse import argparse
import subprocess import subprocess
import torch import torch
......
# Copyright (c) OpenMMLab. All rights reserved. # Copyright (c) OpenMMLab. All rights reserved.
import argparse import argparse
import torch
from collections import OrderedDict from collections import OrderedDict
import torch
def convert_stem(model_key, model_weight, state_dict, converted_names): def convert_stem(model_key, model_weight, state_dict, converted_names):
new_key = model_key.replace('stem.conv', 'conv1') new_key = model_key.replace('stem.conv', 'conv1')
......
# Copyright (c) OpenMMLab. All rights reserved. # Copyright (c) OpenMMLab. All rights reserved.
import argparse import argparse
import mmcv
import os import os
import torch
import warnings import warnings
import mmcv
import torch
from mmcv import Config, DictAction from mmcv import Config, DictAction
from mmcv.cnn import fuse_conv_bn from mmcv.cnn import fuse_conv_bn
from mmcv.parallel import MMDataParallel, MMDistributedDataParallel from mmcv.parallel import MMDataParallel, MMDistributedDataParallel
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment