Commit 44c10a3a authored by Shaoshuai Shi's avatar Shaoshuai Shi
Browse files

sort python imports with isort

parent 2b539530
from pathlib import Path
import subprocess import subprocess
from pathlib import Path
from .version import __version__ from .version import __version__
...@@ -22,4 +22,3 @@ script_version = get_git_commit_number() ...@@ -22,4 +22,3 @@ script_version = get_git_commit_number()
if script_version not in __version__: if script_version not in __version__:
__version__ = __version__ + '+py%s' % script_version __version__ = __version__ + '+py%s' % script_version
from easydict import EasyDict
from pathlib import Path from pathlib import Path
import yaml import yaml
from easydict import EasyDict
def log_config_to_file(cfg, pre='cfg', logger=None): def log_config_to_file(cfg, pre='cfg', logger=None):
......
import torch import torch
from torch.utils.data import DataLoader from torch.utils.data import DataLoader
from torch.utils.data import DistributedSampler as _DistributedSampler
from pcdet.utils import common_utils
from .dataset import DatasetTemplate from .dataset import DatasetTemplate
from .kitti.kitti_dataset import KittiDataset from .kitti.kitti_dataset import KittiDataset
from .nuscenes.nuscenes_dataset import NuScenesDataset from .nuscenes.nuscenes_dataset import NuScenesDataset
from torch.utils.data import DistributedSampler as _DistributedSampler
from pcdet.utils import common_utils
__all__ = { __all__ = {
'DatasetTemplate': DatasetTemplate, 'DatasetTemplate': DatasetTemplate,
......
import numpy as np import numpy as np
from ...utils import common_utils from ...utils import common_utils
......
from functools import partial from functools import partial
import numpy as np import numpy as np
from . import augmentor_utils, database_sampler
from ...utils import common_utils from ...utils import common_utils
from . import augmentor_utils, database_sampler
class DataAugmentor(object): class DataAugmentor(object):
......
import numpy as np
import pickle import pickle
from ...utils import box_utils
import numpy as np
from ...ops.iou3d_nms import iou3d_nms_utils from ...ops.iou3d_nms import iou3d_nms_utils
from ...utils import box_utils
class DataBaseSampler(object): class DataBaseSampler(object):
......
from pathlib import Path
from collections import defaultdict from collections import defaultdict
from pathlib import Path
import numpy as np import numpy as np
import torch.utils.data as torch_data import torch.utils.data as torch_data
from ..utils import common_utils
from .augmentor.data_augmentor import DataAugmentor from .augmentor.data_augmentor import DataAugmentor
from .processor.data_processor import DataProcessor from .processor.data_processor import DataProcessor
from .processor.point_feature_encoder import PointFeatureEncoder from .processor.point_feature_encoder import PointFeatureEncoder
from ..utils import common_utils
class DatasetTemplate(torch_data.Dataset): class DatasetTemplate(torch_data.Dataset):
......
import pickle
import copy import copy
import pickle
import numpy as np import numpy as np
from skimage import io from skimage import io
from ...utils import box_utils, common_utils, calibration_kitti, object3d_kitti
from ..dataset import DatasetTemplate
from ...ops.roiaware_pool3d import roiaware_pool3d_utils from ...ops.roiaware_pool3d import roiaware_pool3d_utils
from ...utils import box_utils, calibration_kitti, common_utils, object3d_kitti
from ..dataset import DatasetTemplate
class KittiDataset(DatasetTemplate): class KittiDataset(DatasetTemplate):
...@@ -435,4 +437,3 @@ if __name__ == '__main__': ...@@ -435,4 +437,3 @@ if __name__ == '__main__':
data_path=ROOT_DIR / 'data' / 'kitti', data_path=ROOT_DIR / 'data' / 'kitti',
save_path=ROOT_DIR / 'data' / 'kitti' save_path=ROOT_DIR / 'data' / 'kitti'
) )
import numpy as np
import numba
import io as sysio import io as sysio
import numba
import numpy as np
from .rotate_iou import rotate_iou_gpu_eval from .rotate_iou import rotate_iou_gpu_eval
......
import time import time
import fire import fire
import .kitti_common as kitti import .kitti_common as kitti
from .eval import get_official_eval_result, get_coco_eval_result from .eval import get_coco_eval_result, get_official_eval_result
def _read_imageset_file(path): def _read_imageset_file(path):
......
...@@ -7,6 +7,7 @@ from collections import OrderedDict ...@@ -7,6 +7,7 @@ from collections import OrderedDict
import numpy as np import numpy as np
from skimage import io from skimage import io
def get_image_index_str(img_idx): def get_image_index_str(img_idx):
return "{:06d}".format(img_idx) return "{:06d}".format(img_idx)
......
...@@ -9,6 +9,7 @@ import numba ...@@ -9,6 +9,7 @@ import numba
import numpy as np import numpy as np
from numba import cuda from numba import cuda
@numba.jit(nopython=True) @numba.jit(nopython=True)
def div_up(m, n): def div_up(m, n):
return m // n + (m % n > 0) return m // n + (m % n > 0)
......
import pickle
import copy import copy
import pickle
from pathlib import Path
import numpy as np import numpy as np
from tqdm import tqdm from tqdm import tqdm
from pathlib import Path
from ...ops.roiaware_pool3d import roiaware_pool3d_utils
from ...utils import common_utils from ...utils import common_utils
from ..dataset import DatasetTemplate from ..dataset import DatasetTemplate
from ...ops.roiaware_pool3d import roiaware_pool3d_utils
class NuScenesDataset(DatasetTemplate): class NuScenesDataset(DatasetTemplate):
......
...@@ -3,15 +3,15 @@ The NuScenes data pre-processing and evaluation is modified from ...@@ -3,15 +3,15 @@ The NuScenes data pre-processing and evaluation is modified from
https://github.com/traveller59/second.pytorch and https://github.com/poodarchu/Det3D https://github.com/traveller59/second.pytorch and https://github.com/poodarchu/Det3D
""" """
from pathlib import Path
import tqdm
import numpy as np
import operator import operator
from functools import reduce from functools import reduce
from pathlib import Path
import numpy as np
import tqdm
from nuscenes.utils.data_classes import Box
from nuscenes.utils.geometry_utils import transform_matrix from nuscenes.utils.geometry_utils import transform_matrix
from pyquaternion import Quaternion from pyquaternion import Quaternion
from nuscenes.utils.data_classes import Box
map_name_from_general_to_detection = { map_name_from_general_to_detection = {
'human.pedestrian.adult': 'pedestrian', 'human.pedestrian.adult': 'pedestrian',
......
from functools import partial from functools import partial
import numpy as np import numpy as np
from ...utils import box_utils, common_utils from ...utils import box_utils, common_utils
......
import torch
import numpy as np
from collections import namedtuple from collections import namedtuple
import numpy as np
import torch
from .detectors import build_detector from .detectors import build_detector
......
from .base_bev_backbone import BaseBEVBackbone from .base_bev_backbone import BaseBEVBackbone
__all__ = { __all__ = {
'BaseBEVBackbone': BaseBEVBackbone 'BaseBEVBackbone': BaseBEVBackbone
} }
\ No newline at end of file
import numpy as np
import torch import torch
import torch.nn as nn import torch.nn as nn
import numpy as np
class BaseBEVBackbone(nn.Module): class BaseBEVBackbone(nn.Module):
......
from .pointnet2_backbone import PointNet2Backbone, PointNet2MSG
from .spconv_backbone import VoxelBackBone8x, VoxelResBackBone8x from .spconv_backbone import VoxelBackBone8x, VoxelResBackBone8x
from .spconv_unet import UNetV2 from .spconv_unet import UNetV2
from .pointnet2_backbone import PointNet2Backbone, PointNet2MSG
__all__ = { __all__ = {
'VoxelBackBone8x': VoxelBackBone8x, 'VoxelBackBone8x': VoxelBackBone8x,
......
from .voxel_set_abstraction import VoxelSetAbstraction from .voxel_set_abstraction import VoxelSetAbstraction
__all__ = { __all__ = {
'VoxelSetAbstraction': VoxelSetAbstraction 'VoxelSetAbstraction': VoxelSetAbstraction
} }
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