Skip to content

GitLab

Unverified Commit 3d6653f4 authored by djiajunustc's avatar djiajunustc Committed by GitHub
Browse files

Update __init__.py (#609) 

missing kornia. before image_to_tensor
parent 26a16123
Show whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
pcdet/models/__init__.py
View file @ 3d6653f4
...@@ -27,7 +27,7 @@ def load_data_to_gpu(batch_dict): ...@@ -27,7 +27,7 @@ def load_data_to_gpu(batch_dict):
elif key in ['frame_id', 'metadata', 'calib']: elif key in ['frame_id', 'metadata', 'calib']:
continue continue
elif key in ['images']: 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']: elif key in ['image_shape']:
batch_dict[key] = torch.from_numpy(val).int().cuda() batch_dict[key] = torch.from_numpy(val).int().cuda()
else: else:
......
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