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
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):
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:
......
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