切片圖像與張量框的張量張量

Question

我有輸入圖像張量[?, 448, 448, 3]張量和我的網絡預測的形狀爲[?, 4]的包圍盒。我想用包圍盒張量切片我的圖像張量，並將得到的張量重新調整爲固定大小的圖像進行進一步處理。

這是可能的tensorflow（甚至更好，原生在凱拉斯）？我已閱讀相關問題。例如，this和this，但它們不適用於索引張量和原始張量都具有未知第一維度的情況。

任何幫助正確的方向是非常感謝！

Answer 1

你最好的方法應該是使用tf.image.crop_and_resize。從文檔：從所述輸入圖像張量

萃取物作物和雙線性調整大小它們 （可能寬高比變化），以通過 crop_size指定一個共同的輸出大小。這比crop_to_bounding_box操作更普遍，其中 從輸入圖像中提取固定大小的切片，並且不允許 調整大小或寬高比更改。

從位於框中的邊界框位置處定義爲 的位置處的輸入圖像返回帶有作物的張量。裁剪後的框都是 調整大小（雙線性插值）爲固定大小= [crop_height， crop_width]。結果是一個4維張量[num_boxes，crop_height， crop_width，depth]。