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做布尔掩码的主要内容，如果未能解决你的问题，请参考以下文章