We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 00d4484 commit ccf1a36Copy full SHA for ccf1a36
torchvision/transforms/v2/functional/_type_conversion.py
@@ -59,6 +59,6 @@ def cvcuda_to_tensor(cvcuda_img: "cvcuda.Tensor") -> torch.Tensor:
59
cuda_tensor = torch.as_tensor(cvcuda_img.cuda(), device="cuda")
60
if cvcuda_img.ndim != 4:
61
raise ValueError(f"Image should be 4 dimensional. Got {cuda_tensor.ndim} dimensions.")
62
- # Convert to NCHW layout from CVCUDA default NHWC
+ # Convert to NCHW shape from CVCUDA default NHWC
63
img = cuda_tensor.permute(0, 3, 1, 2)
64
return img
0 commit comments