cerebras.modelzoo.data.vision.segmentation.transforms.crop_and_pad_augmentations.get_lbs_for_random_crop#

cerebras.modelzoo.data.vision.segmentation.transforms.crop_and_pad_augmentations.get_lbs_for_random_crop(crop_size, data_shape, margins)[source]#
Parameters
  • crop_size –

  • data_shape – (b,c,x,y(,z)) must be the whole thing!

  • margins –

Returns