Shortcuts

ImageToTensor

class mmcv.transforms.ImageToTensor(keys: dict)[source]

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

Parameters

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

transform(results: dict)dict[source]

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

Returns

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

Return type

dict

Read the Docs v: 2.x
Versions
master
latest
2.x
1.x
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
v1.3.12
v1.3.11
v1.3.10
v1.3.9
v1.3.8
v1.3.7
v1.3.6
v1.3.5
v1.3.4
v1.3.3
v1.3.2
v1.3.1
v1.3.0
Downloads
pdf
html
epub
On Read the Docs
Project Home
Builds

Free document hosting provided by Read the Docs.