Skip to content

Commit

Permalink
transpose tensor instead of nparray
Browse files Browse the repository at this point in the history
  • Loading branch information
ly015 committed Feb 15, 2023
1 parent 82a0677 commit 64e2889
Showing 1 changed file with 3 additions and 3 deletions.
6 changes: 3 additions & 3 deletions mmpose/datasets/transforms/formatting.py
Original file line number Diff line number Diff line change
Expand Up @@ -3,7 +3,7 @@

import numpy as np
import torch
from mmcv.transforms import BaseTransform, to_tensor
from mmcv.transforms import BaseTransform
from mmengine.structures import InstanceData, PixelData
from mmengine.utils import is_seq_of

Expand All @@ -27,8 +27,8 @@ def image_to_tensor(img: Union[np.ndarray,
if isinstance(img, np.ndarray):
if len(img.shape) < 3:
img = np.expand_dims(img, -1)
img = np.ascontiguousarray(img.transpose(2, 0, 1))
tensor = to_tensor(img)

tensor = torch.from_numpy(img).permute(2, 0, 1).contiguous()
else:
assert is_seq_of(img, np.ndarray)
tensor = torch.stack([image_to_tensor(_img) for _img in img])
Expand Down

0 comments on commit 64e2889

Please sign in to comment.