Skip to content

GitLab

"tools/git@developer.sourcefind.cn:OpenDAS/openpcdet.git" did not exist on "beb249e5a5f60afcf151405f3e0444700148859f"
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
Hide 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