加载训练和测试图像的函数，其中需要调用read_png、read_jpg、crop_img和normalize_img函数

def load_image_train(img_path, mask_path):
img = read_png(img_path)
mask = read_jpg(mask_path)
img, mask = crop_img(img, mask)

if tf.random.uniform(()) > 0.5:
img = tf.image.random_flip_left_right(img)
mask = tf.image.random_flip_left_right(mask)

if tf.random.uniform(()) > 0.5:
img = tf.image.random_flip_up_down(img)
mask = tf.image.random_flip_up_down(mask)

img, mask = normalize_img(img, mask)
return img, mask

def load_image_test(img_path):
img = read_png(img_path)
img, _ = crop_img(img)
img, _ = normalize_img(img)
return img

其中，load_image_train函数用于加载训练图像和对其进行数据增强，load_image_test函数用于加载测试图像。在load_image_train函数中，read_png函数用于读取PNG格式的图像，read_jpg函数用于读取JPG格式的标签图像，crop_img函数用于裁剪图像，normalize_img函数用于归一化处理。在load_image_test函数中，只调用了read_png函数、crop_img函数和normalize_img函数。