import tensorflow as tf

                         [1, 0]], dtype=tf.float32)
    masks = tf.stack([mask0, mask1, mask2, mask3, mask4, mask5])
    classes = tf.constant([1, 2, 3, 1, 2, 3], dtype=tf.int32)
    scores = tf.constant([1.0, 0.9, 0.8, 0.95, 0.85, 0.6], dtype=tf.float32)
    (nms_masks1,
     nms_scores1,
