2017-10-15 12 views
1

私は入力画像Tensorの形が[?, 448, 448, 3]で、私のネットワークは形状が[?, 4]のバウンディングボックスを予測しています。私はバウンディングボックスのテンソルで画像のテンソルをスライスし、結果のテンソルを次の処理のために固定サイズの画像に再整形したいと思います。画像をスライスする境界ボックス付きテンソル

これは、テンソルフローで可能ですか(またはKerasでネイティブにより優れていますか)。私は関連する質問を読んだ。たとえば、thisおよびthisですが、インデックステンソルと元テンソルの両方が未知の第1次元の場合には適用されません。

正しい方向への助けがあれば幸いです!

答えて

2

tf.image.crop_and_resizeを使用するのが最善の方法です。

を抽出し、入力画像テンソルから作物や双線形それらに crop_sizeによって指定された共通の出力サイズに (おそらくアスペクト比の変更)をサイズ変更:ドキュメントから。これは、 が入力イメージから固定サイズのスライスを抽出し、 のサイズ変更またはアスペクト比の変更を許可しないcrop_to_bounding_box opよりも一般的です。

囲みボックス内の境界ボックス位置に と定義された位置の入力イメージからテクスチャを返します。クロップされたボックスはすべて、固定サイズ= [crop_height、 crop_width]にリサイズ(バイリニア補間)された です。結果は4次元テンソル[num_boxes、crop_height、 crop_width、depth]です。

関連する問題