Table of Contents

Shortcuts

flash.core.data.transforms.kornia_collate¶

flash.core.data.transforms.kornia_collate(samples)[source]¶

Kornia transforms add batch dimension which need to be removed.

This function removes that dimension and then applies torch.utils.data._utils.collate.default_collate.

Return type: Dict[str, Any]

Read the Docs v: 0.7.1

Versions: latest; stable; 0.7.1; 0.7.0; 0.6.0; 0.5.2; 0.5.1; 0.5.0; 0.4.0; 0.3.2; 0.3.1; 0.3.0; 0.2.3; 0.2.2; 0.2.1; 0.2.0; 0.1.0post1; docs-fix_typing

Downloads

On Read the Docs: Project Home; Builds

Free document hosting provided by Read the Docs.