fix wrong in_channels when with_distance=True in VFE layers (#749)

open-mmlab · Jul 14, 2021 · 78f4562 · 78f4562
1 parent 1db4bae
commit 78f4562
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/mmdet3d/models/voxel_encoders/voxel_encoder.py b/mmdet3d/models/voxel_encoders/voxel_encoder.py
@@ -133,7 +133,7 @@ def __init__(self,
         if with_voxel_center:
             in_channels += 3
         if with_distance:
-            in_channels += 3
+            in_channels += 1
         self.in_channels = in_channels
         self._with_distance = with_distance
         self._with_cluster_center = with_cluster_center
@@ -331,7 +331,7 @@ def __init__(self,
         if with_voxel_center:
             in_channels += 3
         if with_distance:
-            in_channels += 3
+            in_channels += 1
         self.in_channels = in_channels
         self._with_distance = with_distance
         self._with_cluster_center = with_cluster_center