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
Show whitespace changes
Inline
Side-by-side
Showing
20 changed files
with
47 additions
and
32 deletions
+47
-32
pcdet/models/backbones_3d/pfe/voxel_set_abstraction.py
pcdet/models/backbones_3d/pfe/voxel_set_abstraction.py
+2
-2
pcdet/models/backbones_3d/pointnet2_backbone.py
pcdet/models/backbones_3d/pointnet2_backbone.py
+1
-0
pcdet/models/backbones_3d/spconv_backbone.py
pcdet/models/backbones_3d/spconv_backbone.py
+3
-2
pcdet/models/backbones_3d/spconv_unet.py
pcdet/models/backbones_3d/spconv_unet.py
+5
-3
pcdet/models/backbones_3d/vfe/__init__.py
pcdet/models/backbones_3d/vfe/__init__.py
+1
-1
pcdet/models/backbones_3d/vfe/mean_vfe.py
pcdet/models/backbones_3d/vfe/mean_vfe.py
+1
-0
pcdet/models/backbones_3d/vfe/pillar_vfe.py
pcdet/models/backbones_3d/vfe/pillar_vfe.py
+1
-0
pcdet/models/dense_heads/__init__.py
pcdet/models/dense_heads/__init__.py
+4
-4
pcdet/models/dense_heads/anchor_head_multi.py
pcdet/models/dense_heads/anchor_head_multi.py
+3
-2
pcdet/models/dense_heads/anchor_head_single.py
pcdet/models/dense_heads/anchor_head_single.py
+1
-1
pcdet/models/dense_heads/anchor_head_template.py
pcdet/models/dense_heads/anchor_head_template.py
+2
-1
pcdet/models/dense_heads/point_head_box.py
pcdet/models/dense_heads/point_head_box.py
+2
-1
pcdet/models/dense_heads/point_head_simple.py
pcdet/models/dense_heads/point_head_simple.py
+2
-1
pcdet/models/dense_heads/point_head_template.py
pcdet/models/dense_heads/point_head_template.py
+2
-1
pcdet/models/dense_heads/point_intra_part_head.py
pcdet/models/dense_heads/point_intra_part_head.py
+2
-1
pcdet/models/dense_heads/target_assigner/atss_target_assigner.py
...odels/dense_heads/target_assigner/atss_target_assigner.py
+2
-1
pcdet/models/dense_heads/target_assigner/axis_aligned_target_assigner.py
...nse_heads/target_assigner/axis_aligned_target_assigner.py
+3
-4
pcdet/models/detectors/__init__.py
pcdet/models/detectors/__init__.py
+3
-3
pcdet/models/detectors/detector3d_template.py
pcdet/models/detectors/detector3d_template.py
+6
-4
pcdet/models/model_utils/model_nms_utils.py
pcdet/models/model_utils/model_nms_utils.py
+1
-0
No files found.
pcdet/models/backbones_3d/pfe/voxel_set_abstraction.py
View file @
44c10a3a
import
torch
import
torch.nn
as
nn
from
....utils
import
common_utils
from
....ops.pointnet2.pointnet2_stack
import
pointnet2_modules
as
pointnet2_stack_modules
from
....ops.pointnet2.pointnet2_stack
import
pointnet2_utils
as
pointnet2_stack_utils
from
....utils
import
common_utils
def
bilinear_interpolate_torch
(
im
,
x
,
y
):
...
...
@@ -236,4 +237,3 @@ class VoxelSetAbstraction(nn.Module):
batch_dict
[
'point_features'
]
=
point_features
# (BxN, C)
batch_dict
[
'point_coords'
]
=
point_coords
# (BxN, 4)
return
batch_dict
pcdet/models/backbones_3d/pointnet2_backbone.py
View file @
44c10a3a
import
torch
import
torch.nn
as
nn
from
...ops.pointnet2.pointnet2_batch
import
pointnet2_modules
from
...ops.pointnet2.pointnet2_stack
import
pointnet2_modules
as
pointnet2_modules_stack
from
...ops.pointnet2.pointnet2_stack
import
pointnet2_utils
as
pointnet2_utils_stack
...
...
pcdet/models/backbones_3d/spconv_backbone.py
View file @
44c10a3a
import
torch.nn
as
nn
import
spconv
from
functools
import
partial
import
spconv
import
torch.nn
as
nn
def
post_act_block
(
in_channels
,
out_channels
,
kernel_size
,
indice_key
=
None
,
stride
=
1
,
padding
=
0
,
conv_type
=
'subm'
,
norm_fn
=
None
):
...
...
pcdet/models/backbones_3d/spconv_unet.py
View file @
44c10a3a
from
functools
import
partial
import
spconv
import
torch
import
torch.nn
as
nn
import
spconv
from
functools
import
partial
from
.spconv_backbone
import
post_act_block
from
...utils
import
common_utils
from
.spconv_backbone
import
post_act_block
class
SparseBasicBlock
(
spconv
.
SparseModule
):
...
...
pcdet/models/backbones_3d/vfe/__init__.py
View file @
44c10a3a
from
.vfe_template
import
VFETemplate
from
.mean_vfe
import
MeanVFE
from
.pillar_vfe
import
PillarVFE
from
.vfe_template
import
VFETemplate
__all__
=
{
'VFETemplate'
:
VFETemplate
,
...
...
pcdet/models/backbones_3d/vfe/mean_vfe.py
View file @
44c10a3a
import
torch
from
.vfe_template
import
VFETemplate
...
...
pcdet/models/backbones_3d/vfe/pillar_vfe.py
View file @
44c10a3a
import
torch
import
torch.nn
as
nn
import
torch.nn.functional
as
F
from
.vfe_template
import
VFETemplate
...
...
pcdet/models/dense_heads/__init__.py
View file @
44c10a3a
from
.anchor_head_
template
import
AnchorHead
Template
from
.anchor_head_
multi
import
AnchorHead
Multi
from
.anchor_head_single
import
AnchorHeadSingle
from
.point_intra_part_head
import
PointIntraPartOffsetHead
from
.point_head_simple
import
PointHeadSimple
from
.anchor_head_template
import
AnchorHeadTemplate
from
.point_head_box
import
PointHeadBox
from
.anchor_head_multi
import
AnchorHeadMulti
from
.point_head_simple
import
PointHeadSimple
from
.point_intra_part_head
import
PointIntraPartOffsetHead
__all__
=
{
'AnchorHeadTemplate'
:
AnchorHeadTemplate
,
...
...
pcdet/models/dense_heads/anchor_head_multi.py
View file @
44c10a3a
import
numpy
as
np
import
torch
import
torch.nn
as
nn
from
.anchor_head_template
import
AnchorHeadTemplate
from
..backbones_2d
import
BaseBEVBackbone
import
torch
from
.anchor_head_template
import
AnchorHeadTemplate
class
SingleHead
(
BaseBEVBackbone
):
...
...
pcdet/models/dense_heads/anchor_head_single.py
View file @
44c10a3a
import
numpy
as
np
import
torch.nn
as
nn
from
.anchor_head_template
import
AnchorHeadTemplate
...
...
@@ -72,4 +73,3 @@ class AnchorHeadSingle(AnchorHeadTemplate):
data_dict
[
'cls_preds_normalized'
]
=
False
return
data_dict
pcdet/models/dense_heads/anchor_head_template.py
View file @
44c10a3a
import
numpy
as
np
import
torch
import
torch.nn
as
nn
from
...utils
import
box_coder_utils
,
common_utils
,
loss_utils
from
.target_assigner.anchor_generator
import
AnchorGenerator
from
.target_assigner.atss_target_assigner
import
ATSSTargetAssigner
from
.target_assigner.axis_aligned_target_assigner
import
AxisAlignedTargetAssigner
from
...utils
import
box_coder_utils
,
loss_utils
,
common_utils
class
AnchorHeadTemplate
(
nn
.
Module
):
...
...
pcdet/models/dense_heads/point_head_box.py
View file @
44c10a3a
import
torch
from
.point_head_template
import
PointHeadTemplate
from
...utils
import
box_coder_utils
,
box_utils
from
.point_head_template
import
PointHeadTemplate
class
PointHeadBox
(
PointHeadTemplate
):
...
...
pcdet/models/dense_heads/point_head_simple.py
View file @
44c10a3a
import
torch
from
.point_head_template
import
PointHeadTemplate
from
...utils
import
box_utils
from
.point_head_template
import
PointHeadTemplate
class
PointHeadSimple
(
PointHeadTemplate
):
...
...
pcdet/models/dense_heads/point_head_template.py
View file @
44c10a3a
import
torch
import
torch.nn
as
nn
import
torch.nn.functional
as
F
from
...utils
import
loss_utils
,
common_utils
from
...ops.roiaware_pool3d
import
roiaware_pool3d_utils
from
...utils
import
common_utils
,
loss_utils
class
PointHeadTemplate
(
nn
.
Module
):
...
...
pcdet/models/dense_heads/point_intra_part_head.py
View file @
44c10a3a
import
torch
from
...utils
import
box_coder_utils
,
box_utils
from
.point_head_template
import
PointHeadTemplate
from
...utils
import
box_utils
,
box_coder_utils
class
PointIntraPartOffsetHead
(
PointHeadTemplate
):
...
...
pcdet/models/dense_heads/target_assigner/atss_target_assigner.py
View file @
44c10a3a
import
torch
from
....utils
import
common_utils
from
....ops.iou3d_nms
import
iou3d_nms_utils
from
....utils
import
common_utils
class
ATSSTargetAssigner
(
object
):
...
...
pcdet/models/dense_heads/target_assigner/axis_aligned_target_assigner.py
View file @
44c10a3a
import
torch
import
numpy
as
np
from
....utils
import
box_utils
import
torch
from
....ops.iou3d_nms
import
iou3d_nms_utils
from
....utils
import
box_utils
class
AxisAlignedTargetAssigner
(
object
):
...
...
@@ -210,5 +211,3 @@ class AxisAlignedTargetAssigner(object):
'reg_weights'
:
reg_weights
,
}
return
ret_dict
pcdet/models/detectors/__init__.py
View file @
44c10a3a
from
.detector3d_template
import
Detector3DTemplate
from
.second_net
import
SECONDNet
from
.PartA2_net
import
PartA2Net
from
.pv_rcnn
import
PVRCNN
from
.pointpillar
import
PointPillar
from
.point_rcnn
import
PointRCNN
from
.pointpillar
import
PointPillar
from
.pv_rcnn
import
PVRCNN
from
.second_net
import
SECONDNet
__all__
=
{
'Detector3DTemplate'
:
Detector3DTemplate
,
...
...
pcdet/models/detectors/detector3d_template.py
View file @
44c10a3a
import
torch
import
os
import
torch
import
torch.nn
as
nn
from
..
import
backbones_3d
,
backbones_2d
,
dense_heads
,
roi_heads
from
..backbones_3d
import
vfe
,
pfe
from
...ops.iou3d_nms
import
iou3d_nms_utils
from
..
import
backbones_2d
,
backbones_3d
,
dense_heads
,
roi_heads
from
..backbones_2d
import
map_to_bev
from
..backbones_3d
import
pfe
,
vfe
from
..model_utils
import
model_nms_utils
from
...ops.iou3d_nms
import
iou3d_nms_utils
class
Detector3DTemplate
(
nn
.
Module
):
...
...
pcdet/models/model_utils/model_nms_utils.py
View file @
44c10a3a
import
torch
from
...ops.iou3d_nms
import
iou3d_nms_utils
...
...
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