solve bc breaking (#1908)

Author: VVsssssk

mmdet3d/models/data_preprocessors/data_preprocessor.py

@@ -13,11 +13,9 @@
 from mmdet3d.registry import MODELS
 from mmdet3d.utils import OptConfigType
 from mmdet.models import DetDataPreprocessor
-from mmdet.models.utils.misc import samplelist_boxtype2tensor
 from .utils import multiview_img_stack_batch
 
 
-
 @MODELS.register_module()
 class Det3DDataPreprocessor(DetDataPreprocessor):
     """Points / Image pre-processor for point clouds / vision-only / multi-
@@ -90,7 +88,6 @@ def __init__(self,
             seg_pad_value=seg_pad_value,
             bgr_to_rgb=bgr_to_rgb,
             rgb_to_bgr=rgb_to_bgr,
-            boxtype2tensor=boxtype2tensor,
             batch_augments=batch_augments)
         self.voxel = voxel
         self.voxel_type = voxel_type
@@ -170,9 +167,14 @@ def simple_process(self, data: dict, training: bool = False) -> dict:
                         'pad_shape': pad_shape
                     })
 
-                if self.boxtype2tensor:
+                if hasattr(self, 'boxtype2tensor') and self.boxtype2tensor:
+                    from mmdet.models.utils.misc import \
+                        samplelist_boxtype2tensor
                     samplelist_boxtype2tensor(data_samples)
-
+                elif hasattr(self, 'boxlist2tensor') and self.boxlist2tensor:
+                    from mmdet.models.utils.misc import \
+                        samplelist_boxlist2tensor
+                    samplelist_boxlist2tensor(data_samples)
                 if self.pad_mask:
                     self.pad_gt_masks(data_samples)
 

mmdet3d/models/layers/spconv/__init__.py

@@ -6,7 +6,9 @@
 except ImportError:
     IS_SPCONV2_AVAILABLE = False
 else:
-    if hasattr(spconv, '__version__') and spconv.__version__ >= '2.0.0':
+    if hasattr(spconv,
+               '__version__') and spconv.__version__ >= '2.0.0' and hasattr(
+                   spconv, 'pytorch'):
         IS_SPCONV2_AVAILABLE = register_spconv2()
     else:
         IS_SPCONV2_AVAILABLE = False