Shortcuts

flash.image.segmentation.input_transform.prepare_target

flash.image.segmentation.input_transform.prepare_target(tensor)[source]

Convert the target mask to long and remove the channel dimension.

Return type

Tensor