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

sort python imports with isort

parent 2b539530
import torch import torch
import torch.nn as nn 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_modules as pointnet2_stack_modules
from ....ops.pointnet2.pointnet2_stack import pointnet2_utils as pointnet2_stack_utils from ....ops.pointnet2.pointnet2_stack import pointnet2_utils as pointnet2_stack_utils
from ....utils import common_utils
def bilinear_interpolate_torch(im, x, y): def bilinear_interpolate_torch(im, x, y):
...@@ -236,4 +237,3 @@ class VoxelSetAbstraction(nn.Module): ...@@ -236,4 +237,3 @@ class VoxelSetAbstraction(nn.Module):
batch_dict['point_features'] = point_features # (BxN, C) batch_dict['point_features'] = point_features # (BxN, C)
batch_dict['point_coords'] = point_coords # (BxN, 4) batch_dict['point_coords'] = point_coords # (BxN, 4)
return batch_dict return batch_dict
import torch import torch
import torch.nn as nn import torch.nn as nn
from ...ops.pointnet2.pointnet2_batch import pointnet2_modules 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_modules as pointnet2_modules_stack
from ...ops.pointnet2.pointnet2_stack import pointnet2_utils as pointnet2_utils_stack from ...ops.pointnet2.pointnet2_stack import pointnet2_utils as pointnet2_utils_stack
......
import torch.nn as nn
import spconv
from functools import partial 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, def post_act_block(in_channels, out_channels, kernel_size, indice_key=None, stride=1, padding=0,
conv_type='subm', norm_fn=None): conv_type='subm', norm_fn=None):
......
from functools import partial
import spconv
import torch import torch
import torch.nn as nn import torch.nn as nn
import spconv
from functools import partial
from .spconv_backbone import post_act_block
from ...utils import common_utils from ...utils import common_utils
from .spconv_backbone import post_act_block
class SparseBasicBlock(spconv.SparseModule): class SparseBasicBlock(spconv.SparseModule):
......
from .vfe_template import VFETemplate
from .mean_vfe import MeanVFE from .mean_vfe import MeanVFE
from .pillar_vfe import PillarVFE from .pillar_vfe import PillarVFE
from .vfe_template import VFETemplate
__all__ = { __all__ = {
'VFETemplate': VFETemplate, 'VFETemplate': VFETemplate,
......
import torch import torch
from .vfe_template import VFETemplate from .vfe_template import VFETemplate
......
import torch import torch
import torch.nn as nn import torch.nn as nn
import torch.nn.functional as F import torch.nn.functional as F
from .vfe_template import VFETemplate from .vfe_template import VFETemplate
......
from .anchor_head_template import AnchorHeadTemplate from .anchor_head_multi import AnchorHeadMulti
from .anchor_head_single import AnchorHeadSingle from .anchor_head_single import AnchorHeadSingle
from .point_intra_part_head import PointIntraPartOffsetHead from .anchor_head_template import AnchorHeadTemplate
from .point_head_simple import PointHeadSimple
from .point_head_box import PointHeadBox 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__ = { __all__ = {
'AnchorHeadTemplate': AnchorHeadTemplate, 'AnchorHeadTemplate': AnchorHeadTemplate,
......
import numpy as np import numpy as np
import torch
import torch.nn as nn import torch.nn as nn
from .anchor_head_template import AnchorHeadTemplate
from ..backbones_2d import BaseBEVBackbone from ..backbones_2d import BaseBEVBackbone
import torch from .anchor_head_template import AnchorHeadTemplate
class SingleHead(BaseBEVBackbone): class SingleHead(BaseBEVBackbone):
......
import numpy as np import numpy as np
import torch.nn as nn import torch.nn as nn
from .anchor_head_template import AnchorHeadTemplate from .anchor_head_template import AnchorHeadTemplate
...@@ -72,4 +73,3 @@ class AnchorHeadSingle(AnchorHeadTemplate): ...@@ -72,4 +73,3 @@ class AnchorHeadSingle(AnchorHeadTemplate):
data_dict['cls_preds_normalized'] = False data_dict['cls_preds_normalized'] = False
return data_dict return data_dict
import numpy as np import numpy as np
import torch import torch
import torch.nn as nn 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.anchor_generator import AnchorGenerator
from .target_assigner.atss_target_assigner import ATSSTargetAssigner from .target_assigner.atss_target_assigner import ATSSTargetAssigner
from .target_assigner.axis_aligned_target_assigner import AxisAlignedTargetAssigner from .target_assigner.axis_aligned_target_assigner import AxisAlignedTargetAssigner
from ...utils import box_coder_utils, loss_utils, common_utils
class AnchorHeadTemplate(nn.Module): class AnchorHeadTemplate(nn.Module):
......
import torch import torch
from .point_head_template import PointHeadTemplate
from ...utils import box_coder_utils, box_utils from ...utils import box_coder_utils, box_utils
from .point_head_template import PointHeadTemplate
class PointHeadBox(PointHeadTemplate): class PointHeadBox(PointHeadTemplate):
......
import torch import torch
from .point_head_template import PointHeadTemplate
from ...utils import box_utils from ...utils import box_utils
from .point_head_template import PointHeadTemplate
class PointHeadSimple(PointHeadTemplate): class PointHeadSimple(PointHeadTemplate):
......
import torch import torch
import torch.nn as nn import torch.nn as nn
import torch.nn.functional as F import torch.nn.functional as F
from ...utils import loss_utils, common_utils
from ...ops.roiaware_pool3d import roiaware_pool3d_utils from ...ops.roiaware_pool3d import roiaware_pool3d_utils
from ...utils import common_utils, loss_utils
class PointHeadTemplate(nn.Module): class PointHeadTemplate(nn.Module):
......
import torch import torch
from ...utils import box_coder_utils, box_utils
from .point_head_template import PointHeadTemplate from .point_head_template import PointHeadTemplate
from ...utils import box_utils, box_coder_utils
class PointIntraPartOffsetHead(PointHeadTemplate): class PointIntraPartOffsetHead(PointHeadTemplate):
......
import torch import torch
from ....utils import common_utils
from ....ops.iou3d_nms import iou3d_nms_utils from ....ops.iou3d_nms import iou3d_nms_utils
from ....utils import common_utils
class ATSSTargetAssigner(object): class ATSSTargetAssigner(object):
......
import torch
import numpy as np import numpy as np
from ....utils import box_utils import torch
from ....ops.iou3d_nms import iou3d_nms_utils from ....ops.iou3d_nms import iou3d_nms_utils
from ....utils import box_utils
class AxisAlignedTargetAssigner(object): class AxisAlignedTargetAssigner(object):
...@@ -210,5 +211,3 @@ class AxisAlignedTargetAssigner(object): ...@@ -210,5 +211,3 @@ class AxisAlignedTargetAssigner(object):
'reg_weights': reg_weights, 'reg_weights': reg_weights,
} }
return ret_dict return ret_dict
from .detector3d_template import Detector3DTemplate from .detector3d_template import Detector3DTemplate
from .second_net import SECONDNet
from .PartA2_net import PartA2Net from .PartA2_net import PartA2Net
from .pv_rcnn import PVRCNN
from .pointpillar import PointPillar
from .point_rcnn import PointRCNN from .point_rcnn import PointRCNN
from .pointpillar import PointPillar
from .pv_rcnn import PVRCNN
from .second_net import SECONDNet
__all__ = { __all__ = {
'Detector3DTemplate': Detector3DTemplate, 'Detector3DTemplate': Detector3DTemplate,
......
import torch
import os import os
import torch
import torch.nn as nn 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_2d import map_to_bev
from ..backbones_3d import pfe, vfe
from ..model_utils import model_nms_utils from ..model_utils import model_nms_utils
from ...ops.iou3d_nms import iou3d_nms_utils
class Detector3DTemplate(nn.Module): class Detector3DTemplate(nn.Module):
......
import torch import torch
from ...ops.iou3d_nms import iou3d_nms_utils from ...ops.iou3d_nms import iou3d_nms_utils
......
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