Shortcuts

ImageToTensor

class mmcv.transforms.ImageToTensor(keys: dict)[源代码]

Convert image to torch.Tensor by given keys.

The dimension order of input image is (H, W, C). The pipeline will convert it to (C, H, W). If only 2 dimension (H, W) is given, the output would be (1, H, W).

Required keys:

  • all these keys in keys

Modified Keys:

  • all these keys in keys

参数

keys (Sequence[str]) – Key of images to be converted to Tensor.

transform(results: dict)dict[源代码]

Transform function to convert image in results to torch.Tensor and transpose the channel order. :param results: Result dict contains the image data to convert. :type results: dict

返回

The result dict contains the image converted to :obj:torch.Tensor and transposed to (C, H, W) order.

返回类型

dict

Read the Docs v: latest
Versions
latest
stable
2.x
v2.1.0
v2.0.1
v2.0.0
1.x
v1.7.1
v1.7.0
v1.6.2
v1.6.1
v1.6.0
v1.5.3
v1.5.2_a
v1.5.1
v1.5.0
v1.4.8
v1.4.7
v1.4.6
v1.4.5
v1.4.4
v1.4.3
v1.4.2
v1.4.1
v1.4.0
v1.3.18
v1.3.17
v1.3.16
v1.3.15
v1.3.14
v1.3.13
Downloads
pdf
html
epub
On Read the Docs
Project Home
Builds

Free document hosting provided by Read the Docs.