UNET预测

Question

我使用unet进行图像分割，我的问题是下面的代码意味着什么？

test_img_norm=test_img[:,:,0][:,:,None] 

和

prediction_other = (model.predict(test_img_other_input)[0,:,:,0] > 0.2).astype(np.uint8)

Answer 1

关于第一个问题test_img_norm=test_img[:,:,0][:,:,None]，test_img[:,:,0]将复制图像的第一个通道，test_img[:,:,0][:,:,None]将为其添加一个通道。例如，如果您有一个形状为(256, 256, 3)的图像，则test_img_norm形状将为(256, 256, 1)。

关于问题的第二部分，model.predict(test_img_other_input)[0,:,:,0] > 0.2将给出一个布尔数组。对于UNet输出中的每个元素，如果元素小于0.2，则输出为True，否则为False。最后，.astype(np.uint8)将布尔人变成0或1。