tensorflow - 通过在张量流中提供像素标签来获取要裁剪的对象的位置

我有一个图像数据集（每个图像都是 rgb 格式）和相应的标签图像（包含图像中每个像素的标签）。我需要从原始图像中提取特定类的对象（像素）。

首先我必须使用标签图像找到对象的位置（通过提供给定对象的标签）（通过使用显式 for 循环是可行的，但是我不想使用显式 for 循环）

现在我的问题-

如果张量流中有任何内置函数可以为我提供给定对象的位置（矩形很好）（如果我提供该对象的标签）？

之后，我可以使用 tf.image.crop_and_resize 裁剪图像。但我找不到任何可以为我提供对象位置的功能。

标签： tensorflow