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

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

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

Returns