Update __init__.py (#609)

missing kornia. before image_to_tensor

pcdet/models/__init__.py

@@ -27,7 +27,7 @@ def load_data_to_gpu(batch_dict):
         elif key in ['frame_id', 'metadata', 'calib']:
             continue
         elif key in ['images']:
-            batch_dict[key] = image_to_tensor(val).float().cuda().contiguous()
+            batch_dict[key] = kornia.image_to_tensor(val).float().cuda().contiguous()
         elif key in ['image_shape']:
             batch_dict[key] = torch.from_numpy(val).int().cuda()
         else: