crop_img函数定义

以下是一个示例的crop_img函数定义：

def crop_img(img, mask=None, crop_size=256):
    height, width = img.shape[:2]
    if height <= crop_size or width <= crop_size:
        return img, mask
    
    x = tf.random.uniform(shape=[], minval=0, maxval=width-crop_size, dtype=tf.int32)
    y = tf.random.uniform(shape=[], minval=0, maxval=height-crop_size, dtype=tf.int32)
    
    img = img[y:y+crop_size, x:x+crop_size]
    
    if mask is not None:
        mask = mask[y:y+crop_size, x:x+crop_size]
    
    return img, mask

这个crop_img函数接受输入的图像img和标签图像mask（可选），以及裁剪尺寸crop_size。函数首先获取输入图像的高度和宽度，如果高度或宽度小于等于裁剪尺寸，则直接返回图像和标签图像（如果有）。否则，函数会生成一个随机的裁剪位置，并使用该位置对图像进行裁剪，裁剪尺寸为crop_size。如果存在标签图像，则也对标签图像进行相同的裁剪。最后返回裁剪后的图像和标签图像（如果有）。