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