我有一个图像数据集（每个图像都是rgb格式）和相应的标签图像（其中包含图像中每个像素的标签）。 我需要从原始图像中提取特定类别的对象（像素）。
首先我必须使用标签图像（通过提供给定对象的标签）找到对象的位置（通过使用显式的for循环是可行的，但我不想使用显式的for循环）
现在我的问题-
如果tensorflow中有任何内置函数可以给我指定对象的位置（矩形可以）（如果我提供该对象的标签）？
之后，我可以使用tf.image.crop_and_resize裁剪图像。但是我找不到任何可以给我定位对象的函数。