flash.image.segmentation.input_transform.prepare_target¶

flash.image.segmentation.input_transform.prepare_target(batch)[source]¶

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