首先,抱歉标题含糊
由于我有兴趣了解有关 TensorFlow 和图像分割的更多信息,因此我正在关注他们的教程 ( https://www.tensorflow.org/tutorials/images/segmentation )。但是,我注意到一些我无法完全掌握的东西,也不是在谷歌搜索之后。
在这个部分:
def create_mask(pred_mask):
pred_mask = tf.argmax(pred_mask, axis=-1)
pred_mask = pred_mask[..., tf.newaxis]
return pred_mask[0]
首先为 pred_mask 向量创建一个新轴,然后只选择第一个元素的原因是什么?为什么不像我预期的那样,如下图所示:
def create_mask(pred_mask):
pred_mask = tf.argmax(pred_mask, axis=-1)
return pred_mask