Skip to content

incompatible dtype and options in padding #4747

@wyli

Description

@wyli

Describe the bug

from monai.transforms import RandZoomd, CastToTyped
import numpy as np
import torch

a = torch.zeros(1, 10, 10)
RandZoomd("image", prob=1.0, max_zoom=1.0, mode="nearest")(CastToTyped("image", dtype=np.uint8)({"image": a}))

output:

Traceback (most recent call last):
  File "my_test_1.py", line 6, in <module>
    RandZoomd("image", prob=1.0, max_zoom=1.0, mode="nearest")(CastToTyped("image", dtype=np.uint8)({"image": a}))
  File "/Users/Documents/MONAI/monai/transforms/spatial/dictionary.py", line 1583, in __call__
    d[key] = self.rand_zoom(
  File "/Users/Documents/MONAI/monai/transforms/spatial/array.py", line 1656, in __call__
    out = Zoom(
  File "/Users/Documents/MONAI/monai/transforms/spatial/array.py", line 1162, in __call__
    out = _pad_crop(out)
  File "/Users/Documents/MONAI/monai/transforms/croppad/array.py", line 1304, in __call__
    ret = self.padder(self.cropper(img), mode=mode, **pad_kwargs)
  File "/Users/Documents/MONAI/monai/transforms/croppad/array.py", line 171, in __call__
    out = self._pt_pad(img_t, pad_width=to_pad_, mode=mode_, **kwargs_)
  File "/Users/Documents/MONAI/monai/transforms/croppad/array.py", line 134, in _pt_pad
    return pad_pt(img.unsqueeze(0), pt_pad_width, mode=mode, **kwargs).squeeze(0)
  File "/Users/Documents/MONAI/monai/data/meta_tensor.py", line 245, in __torch_function__
    ret = super().__torch_function__(func, types, args, kwargs)
  File "/usr/local/anaconda3/envs/py38/lib/python3.8/site-packages/torch/_tensor.py", line 1121, in __torch_function__
    ret = func(*args, **kwargs)
RuntimeError: "replication_pad2d_cpu" not implemented for 'Byte'

Metadata

Metadata

Assignees

Labels

No labels
No labels

Type

No type

Projects

No projects

Milestone

No milestone

Relationships

None yet

Development

No branches or pull requests

Issue actions