tensorflow 2.0 中有 cv2 findcontours 和 boundingRect 吗？

Question

这与 cv2.findContours 和 cv2.boundingRect 的功能相同，我尝试使用谷歌搜索并查看 tf.image.* 但它似乎没有实现。有什么建议吗？我喜欢在一批二值化图像中执行此操作，并输出形状为 (batch_num, n_boxes, 4) 的边界矩形，最后一个暗淡对应于框的 (x1, y1, x2, y2)。目前，如果我有一批图像，我可以使用 cv2.findContours(...) 后跟 cv2.boundingRect(...) 和 for 循环来完成此操作。我的目标是在纯 tf（图形？）中完成这项工作，这样它就可以成为训练或推理工作流程的一部分。如果我无法将其集成到 TF 中，则性能太慢。

Answer 1

Tensorflow 旨在构建计算图来表示一系列操作，以便我们最终提出损失函数。我们反向支持这个损失函数，这样我们就可以最小化权重。 cv2.findContours 和 cv2.boundingRect 是由于算法本身的性质而无法在 GPU 上并行化的操作。 tf.image 中的任何内容都可以很容易地表示为计算图。无法以这种方式在图像中查找轮廓，因此该功能自然不可用。您别无选择，只能使用for 循环，但由于这完全在 CPU 上完成，您也许可以使用多处理或线程在每个图像上并行执行此操作。