Shortcuts

ToTensor

class mmcv.transforms.ToTensor(keys: Sequence[str])[源代码]

Convert some results to torch.Tensor by given keys.

Required keys:

  • all these keys in keys

Modified Keys:

  • all these keys in keys

参数

keys (Sequence[str]) – Keys that need to be converted to Tensor.

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

Transform function to convert data to torch.Tensor.

参数

results (dict) – Result dict from loading pipeline.

返回

keys in results will be updated.

返回类型

dict

Read the Docs v: stable
Versions
latest
stable
2.x
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.