TensorToArray#
module : eotorchloader.transform.tensor
- class TensorToArray(img_only: bool = False, mask_only: bool = False)#
Convert (image, mask) sample torch tensor to numpy array
Note
tensor in gpu
- Parameters
img_only (str) – Apply transform only to image
mask_only (List(Int)) – Apply transform only to mask
see ToTorchTensor