cerebras.modelzoo.data.vision.segmentation.preprocessing_utils.rotation_90_transform

cerebras.modelzoo.data.vision.segmentation.preprocessing_utils.rotation_90_transform(img, num_rotations)

Function equivalent to tf.image.rot90 Rotates img in counter clockwise direction :params img: torch.Tensor of shape (C, H, W) or (H, W) :params num_rotations: int value representing

number of counter clock-wise rotations of img

previous

cerebras.modelzoo.data.vision.segmentation.preprocessing_utils.resize_image_with_crop_or_pad_transform

next

cerebras.modelzoo.data.vision.segmentation.preprocessing_utils.tile_image_transform