我目前正在使用这个代码来获取一个图像的灰度图像表示,并以
(512, 370, 1)
数组。
img_instance = cv2.imread(df.iloc[i][x_col]) / 255.
img_instance = cv2.resize(img_instance, (target_size[1], target_size[0]))
mask_instance = cv2.imread(df.iloc[i][y_col], cv2.IMREAD_GRAYSCALE) / 255.
mask_instance = cv2.resize(mask_instance, (target_size[1], target_size[0]))
print(mask_instance.shape)
mask_instance.reshape(target_size[0], target_size[1], 1)
print(mask_instance.shape)
(512, 370)
(512, 370)
__________________________________________________________________________________________________
conv2d_19 (Conv2D) (None, 512, 370, 1) 17 activation_18[0][0]