From c976efe7cedc755869e49108dcfd16f7190b9d6c Mon Sep 17 00:00:00 2001 From: vfdev Date: Wed, 25 Sep 2024 17:08:38 +0200 Subject: [PATCH] Fix image segmentation template CI --- src/templates/template-vision-segmentation/data.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/src/templates/template-vision-segmentation/data.py b/src/templates/template-vision-segmentation/data.py index 33326d28..b83000fd 100644 --- a/src/templates/template-vision-segmentation/data.py +++ b/src/templates/template-vision-segmentation/data.py @@ -99,7 +99,7 @@ def setup_data(config: Namespace): transform_train = A.Compose( [ A.RandomScale(scale_limit=(0.0, 1.5), interpolation=cv2.INTER_LINEAR, p=1.0), - A.PadIfNeeded(val_img_size, val_img_size, border_mode=cv2.BORDER_CONSTANT), + A.PadIfNeeded(val_img_size, val_img_size, border_mode=cv2.BORDER_CONSTANT, value=0), A.RandomCrop(train_img_size, train_img_size), A.HorizontalFlip(), A.Blur(blur_limit=3), @@ -111,7 +111,7 @@ def setup_data(config: Namespace): transform_eval = A.Compose( [ - A.PadIfNeeded(val_img_size, val_img_size, border_mode=cv2.BORDER_CONSTANT), + A.PadIfNeeded(val_img_size, val_img_size, border_mode=cv2.BORDER_CONSTANT, value=0), A.Normalize(mean=mean, std=std), ignore_mask_boundaries, ToTensor(),