Skip to content
GitLab
Menu
Projects
Groups
Snippets
Loading...
Help
Help
Support
Community forum
Keyboard shortcuts
?
Submit feedback
Contribute to GitLab
Sign in / Register
Toggle navigation
Menu
Open sidebar
OpenDAS
OpenPCDet
Commits
44c10a3a
Commit
44c10a3a
authored
Jul 30, 2020
by
Shaoshuai Shi
Browse files
sort python imports with isort
parent
2b539530
Changes
68
Hide whitespace changes
Inline
Side-by-side
Showing
20 changed files
with
51 additions
and
32 deletions
+51
-32
pcdet/__init__.py
pcdet/__init__.py
+1
-2
pcdet/config.py
pcdet/config.py
+2
-1
pcdet/datasets/__init__.py
pcdet/datasets/__init__.py
+4
-2
pcdet/datasets/augmentor/augmentor_utils.py
pcdet/datasets/augmentor/augmentor_utils.py
+1
-0
pcdet/datasets/augmentor/data_augmentor.py
pcdet/datasets/augmentor/data_augmentor.py
+3
-1
pcdet/datasets/augmentor/database_sampler.py
pcdet/datasets/augmentor/database_sampler.py
+4
-2
pcdet/datasets/dataset.py
pcdet/datasets/dataset.py
+4
-2
pcdet/datasets/kitti/kitti_dataset.py
pcdet/datasets/kitti/kitti_dataset.py
+5
-4
pcdet/datasets/kitti/kitti_object_eval_python/eval.py
pcdet/datasets/kitti/kitti_object_eval_python/eval.py
+4
-2
pcdet/datasets/kitti/kitti_object_eval_python/evaluate.py
pcdet/datasets/kitti/kitti_object_eval_python/evaluate.py
+2
-1
pcdet/datasets/kitti/kitti_object_eval_python/kitti_common.py
...t/datasets/kitti/kitti_object_eval_python/kitti_common.py
+1
-0
pcdet/datasets/kitti/kitti_object_eval_python/rotate_iou.py
pcdet/datasets/kitti/kitti_object_eval_python/rotate_iou.py
+1
-0
pcdet/datasets/nuscenes/nuscenes_dataset.py
pcdet/datasets/nuscenes/nuscenes_dataset.py
+5
-3
pcdet/datasets/nuscenes/nuscenes_utils.py
pcdet/datasets/nuscenes/nuscenes_utils.py
+5
-5
pcdet/datasets/processor/data_processor.py
pcdet/datasets/processor/data_processor.py
+2
-0
pcdet/models/__init__.py
pcdet/models/__init__.py
+4
-2
pcdet/models/backbones_2d/__init__.py
pcdet/models/backbones_2d/__init__.py
+1
-2
pcdet/models/backbones_2d/base_bev_backbone.py
pcdet/models/backbones_2d/base_bev_backbone.py
+1
-1
pcdet/models/backbones_3d/__init__.py
pcdet/models/backbones_3d/__init__.py
+1
-1
pcdet/models/backbones_3d/pfe/__init__.py
pcdet/models/backbones_3d/pfe/__init__.py
+0
-1
No files found.
pcdet/__init__.py
View file @
44c10a3a
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
pcdet/config.py
View file @
44c10a3a
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
):
...
...
pcdet/datasets/__init__.py
View file @
44c10a3a
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
,
...
...
pcdet/datasets/augmentor/augmentor_utils.py
View file @
44c10a3a
import
numpy
as
np
import
numpy
as
np
from
...utils
import
common_utils
from
...utils
import
common_utils
...
...
pcdet/datasets/augmentor/data_augmentor.py
View file @
44c10a3a
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
):
...
...
pcdet/datasets/augmentor/database_sampler.py
View file @
44c10a3a
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
):
...
...
pcdet/datasets/dataset.py
View file @
44c10a3a
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
):
...
...
pcdet/datasets/kitti/kitti_dataset.py
View file @
44c10a3a
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'
)
)
pcdet/datasets/kitti/kitti_object_eval_python/eval.py
View file @
44c10a3a
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
...
...
pcdet/datasets/kitti/kitti_object_eval_python/evaluate.py
View file @
44c10a3a
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
):
...
...
pcdet/datasets/kitti/kitti_object_eval_python/kitti_common.py
View file @
44c10a3a
...
@@ -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
)
...
...
pcdet/datasets/kitti/kitti_object_eval_python/rotate_iou.py
View file @
44c10a3a
...
@@ -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
)
...
...
pcdet/datasets/nuscenes/nuscenes_dataset.py
View file @
44c10a3a
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
):
...
...
pcdet/datasets/nuscenes/nuscenes_utils.py
View file @
44c10a3a
...
@@ -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'
,
...
...
pcdet/datasets/processor/data_processor.py
View file @
44c10a3a
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
...
...
pcdet/models/__init__.py
View file @
44c10a3a
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
...
...
pcdet/models/backbones_2d/__init__.py
View file @
44c10a3a
from
.base_bev_backbone
import
BaseBEVBackbone
from
.base_bev_backbone
import
BaseBEVBackbone
__all__
=
{
__all__
=
{
'BaseBEVBackbone'
:
BaseBEVBackbone
'BaseBEVBackbone'
:
BaseBEVBackbone
}
}
\ No newline at end of file
pcdet/models/backbones_2d/base_bev_backbone.py
View file @
44c10a3a
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
):
...
...
pcdet/models/backbones_3d/__init__.py
View file @
44c10a3a
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
,
...
...
pcdet/models/backbones_3d/pfe/__init__.py
View file @
44c10a3a
from
.voxel_set_abstraction
import
VoxelSetAbstraction
from
.voxel_set_abstraction
import
VoxelSetAbstraction
__all__
=
{
__all__
=
{
'VoxelSetAbstraction'
:
VoxelSetAbstraction
'VoxelSetAbstraction'
:
VoxelSetAbstraction
}
}
Prev
1
2
3
4
Next
Write
Preview
Markdown
is supported
0%
Try again
or
attach a new file
.
Attach a file
Cancel
You are about to add
0
people
to the discussion. Proceed with caution.
Finish editing this message first!
Cancel
Please
register
or
sign in
to comment