【发布时间】:2021-08-14 08:36:35
【问题描述】:
有人可以帮我在语义分割任务中,应该在批处理生成器类中对图像和掩码进行归一化还是只对其中一个进行归一化?
我正在使用以下代码来规范化图像和蒙版:
mean_val, std_val = img.mean(), img.std()
img = (img - mean_val)/std_val
例如:
这里的图像和相应的掩码针对前列腺癌分割任务进行了归一化
这里只有掩码被标准化
def __getitem__(self,i):
index= self.indexes[i * self.batch_size : (i + 1) * self.batch_size]
X = np.empty((self.batch_size, self.crop_dim[0], self.crop_dim[1],3)).astype(np.uint8)
Y = np.empty((self.batch_size, self.crop_dim[0], self.crop_dim[1],5)).astype(np.uint8)
for i,ID in enumerate(index):
dim= (self.crop_dim[0],self.crop_dim[1])
img=cv2.imread(self.img_list[ID],cv2.COLOR_BGR2RGB)
img = cv2.resize(img,dim)
mask=imageio.imread(self.labels[ID],as_gray=False, pilmode="RGB")
mask = cv2.resize(mask,dim)
mask= create_labels(mask)
# Augement training patches only
if self.augmentation:
sample = self.augmentation(image=img_numpy, mask=mask_numpy)
img_numpy, mask_numpy = sample['image'], sample['mask']
mean_val, std_val = img.mean(), img.std()
img = (img - mean_val)/std_val
mean_val_mask, std_val_mask = mask.mean(), mask.std()
mask= (mask - mean_val_mask)/std_val_mask
X[i,]=img_numpy
Y[i,]=mask_numpy
【问题讨论】:
标签: tensorflow opencv keras image-segmentation semantic-segmentation