mirror of
https://github.com/JDAI-CV/fast-reid.git
synced 2025-06-03 14:50:47 +08:00
fix docs in ToTensor
This commit is contained in:
parent
896933bf5e
commit
96b9ad2d99
@ -20,7 +20,7 @@ class ToTensor(object):
|
||||
"""Convert a ``PIL Image`` or ``numpy.ndarray`` to tensor.
|
||||
|
||||
Converts a PIL Image or numpy.ndarray (H x W x C) in the range
|
||||
[0, 255] to a torch.FloatTensor of shape (C x H x W) in the range [0.0, 1.0]
|
||||
[0, 255] to a torch.FloatTensor of shape (C x H x W) in the range [0.0, 255.0]
|
||||
if the PIL Image belongs to one of the modes (L, LA, P, I, F, RGB, YCbCr, RGBA, CMYK, 1)
|
||||
or if the numpy.ndarray has dtype = np.uint8
|
||||
|
||||
|
Loading…
x
Reference in New Issue
Block a user