Skip to content

GitLab

Unverified Commit 78f45621 authored by Ziyi Wu's avatar Ziyi Wu Committed by GitHub
Browse files

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

parent 1db4baed
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 2 deletions
mmdet3d/models/voxel_encoders/voxel_encoder.py
View file @ 78f45621
...@@ -133,7 +133,7 @@ class DynamicVFE(nn.Module): ...@@ -133,7 +133,7 @@ class DynamicVFE(nn.Module):
if with_voxel_center: if with_voxel_center:
in_channels += 3 in_channels += 3
if with_distance: if with_distance:
in_channels += 3 in_channels += 1
self.in_channels = in_channels self.in_channels = in_channels
self._with_distance = with_distance self._with_distance = with_distance
self._with_cluster_center = with_cluster_center self._with_cluster_center = with_cluster_center
...@@ -331,7 +331,7 @@ class HardVFE(nn.Module): ...@@ -331,7 +331,7 @@ class HardVFE(nn.Module):
if with_voxel_center: if with_voxel_center:
in_channels += 3 in_channels += 3
if with_distance: if with_distance:
in_channels += 3 in_channels += 1
self.in_channels = in_channels self.in_channels = in_channels
self._with_distance = with_distance self._with_distance = with_distance
self._with_cluster_center = with_cluster_center self._with_cluster_center = with_cluster_center
......
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