wrong index when adjusting z-wise

[170928]

In function ' coarse2fine_localization', for adjusting z-wise only, I think we use cam_maps_3d[:, :, :, :, 0], not [:, :, :, :, 2] because self.coordmap_3d is the result of concatenate (Z_map, Y_map, X_map).

        center_maps_3d = self.center_map_refiner(center_maps_3d.unsqueeze(1)).squeeze(1)
        cam_maps_3d = self.coordmap_3d + cam_maps_offset.unsqueeze(-1).transpose(4,1).contiguous()
        cam_maps_3d[:,:,:,:,2] = cam_maps_3d[:,:,:,:,2] + cam_maps_offset_bv.unsqueeze(2).contiguous()

[Arthur151]

Hi, MJ Shin @170928 Thanks a lot for pointing out this error! I will fix it in the next version.

[vivi90]

@Arthur151 Is it already fixed? 🙂

[Arthur151]

Yes.