2017-03-23 23 views
1

フィーチャマップのペアのバッチ全体で2Dクロスコンボリューション(例えば、MATLABのhttps://www.mathworks.com/help/signal/ref/xcorr2.htmlのように)を実行する必要があります。明確にするためTensorflowにおけるバッチ相互相関

XB x W1 x H1 x CこととYB x W2 x H2 x Cとします。

欲しい出力は、(我々が同じパディングとY上を通過する「フィルタ」としてXを治療すると仮定して)形状B x W2 x H2 x 1である出力のi1 x W2 x H2 x 1番目のスライスがX[i,:,:,:]Y[i,:,:,:]例えば間の相互相関であります

tf.nn.conv2d(Y[i,:,:,:], X[i,:,:,:], [1,1,1,1], padding='SAME') 

のようなものは、この動作を実現する効率的な方法はありますか?

注:X1 x W1 x H1 x Cであり、我々は各スライスBYとそれをクロス相関したい場合、これは簡単です:

相互相関としてTensorflowが conv2dを実装しているという事実を利用しています
cross_corr = tf.nn.conv2d(
    Y, tf.transpose(X, perm[1,2,3,0], [1,1,1,1], padding='SAME') 

小さなテンソルを本質的に転置後のフィルターとして扱うことができるという事実がある。 B異なるフィルタのクロスコンボリューションを取る必要があるので、これは私の問題を解決しません。

多分conv3dは可能でしょうか?

注2:matconvnetのvl_nnconvは、フィルタチャネルが入力チャネルを分割する場合にこれを行います。 Tensorflowには同等の機能がありますか?

答えて

0

hereのようにtf.mapの機能を使用できます。

conv3dを使用してみましたが、動作させる方法が見つかりませんでした。

+0

答えてくれてありがとう、これが私が探しているものです。 'tf.expand_dims(inputs [0]、0)'を呼び出すと、ちょうど1つのコメント(私は元の答えにコメントするのに十分な担当者がいません)が、余分なディメンションを追加しませんか? tf.nn.conv2dの入力を5Dテンソルにしないでください(これは可能ですか?) また、万が一、これを実行しようとしましたか?マップ関数は実際にGPUで並列化されていますか? –

+0

私は走ってみました、それは動作します。スピードについては、正直に考えていない。 基本的にマップ(4D、4D)は、各(3D、3D)スライス上に関数を適用します。基本的には「画像ごと」の関数です。 2つの 'tf.expand_dims'は、(3D、3D)を(4D、4D)に変換して、互いに絡み合っています。 最終的な結果は5Dテンソルですが、2次元が1に等しいので、最後にそれらを取り除きます。 (私は明確化のために私の元の答えをコメントしました) – Evariste