python TensorFlow做布尔掩码

Posted

tags:

篇首语:本文由小常识网(cha138.com)小编为大家整理,主要介绍了python TensorFlow做布尔掩码相关的知识,希望对你有一定的参考价值。

import tensorflow as tf
above_threshold = (box_class_scores >= threshold) # shape: (19, 19, 5)
scores = tf.boolean_mask(box_class_scores, above_threshold)

以上是关于python TensorFlow做布尔掩码的主要内容,如果未能解决你的问题,请参考以下文章

如何为图像分割创建带有掩码的自定义图像数据集?(特别是对于 Tensorflow)

按模式查找布尔掩码

1.6 比较掩码和布尔逻辑

通过布尔掩码数组选择numpy数组的元素

使用 Tensorflow 数据集解码 RLE(运行长度编码)掩码

TenSorFlow-掩码