切片图像与张量框的张量张量
问题描述:
我有输入图像张量[?, 448, 448, 3]
张量和我的网络预测的形状为[?, 4]
的包围盒。我想用包围盒张量切片我的图像张量,并将得到的张量重新调整为固定大小的图像进行进一步处理。切片图像与张量框的张量张量
这是可能的tensorflow(甚至更好,原生在凯拉斯)?我已阅读相关问题。例如,this和this,但它们不适用于索引张量和原始张量都具有未知第一维度的情况。
任何帮助正确的方向是非常感谢!
答
你最好的方法应该是使用tf.image.crop_and_resize
。从文档:从所述输入图像张量
萃取物作物和双线性调整大小它们 (可能宽高比变化),以通过 crop_size指定一个共同的输出大小。这比crop_to_bounding_box操作更普遍,其中 从输入图像中提取固定大小的切片,并且不允许 调整大小或宽高比更改。
从位于框中的边界框位置处定义为 的位置处的输入图像返回带有作物的张量。裁剪后的框都是 调整大小(双线性插值)为固定大小= [crop_height, crop_width]。结果是一个4维张量[num_boxes,crop_height, crop_width,depth]。