切片图像与张量框的张量张量

问题描述:

我有输入图像张量[?, 448, 448, 3]张量和我的网络预测的形状为[?, 4]的包围盒。我想用包围盒张量切片我的图像张量,并将得到的张量重新调整为固定大小的图像进行进一步处理。切片图像与张量框的张量张量

这是可能的tensorflow(甚至更好,原生在凯拉斯)?我已阅读相关问题。例如,thisthis,但它们不适用于索引张量和原始张量都具有未知第一维度的情况。

任何帮助正确的方向是非常感谢!

你最好的方法应该是使用tf.image.crop_and_resize。从文档:从所述输入图像张量

萃取物作物和双线性调整大小它们 (可能宽高比变化),以通过 crop_size指定一个共同的输出大小。这比crop_to_bounding_box操作更普遍,其中 从输入图像中提取固定大小的切片,并且不允许 调整大小或宽高比更改。

从位于框中的边界框位置处定义为 的位置处的输入图像返回带有作物的张量。裁剪后的框都是 调整大小(双线性插值)为固定大小= [crop_height, crop_width]。结果是一个4维张量[num_boxes,crop_height, crop_width,depth]。