2016-04-25 16 views
7

Tamamen konvolüsyonel bir ağ gibi bir şey uygulamaya çalışıyorum; burada son dönüşüm katmanı 1x1 filtre boyutunu kullanır ve bir 'skor' tensörü çıkarır. Skor tensörü şekli vardır [Toplu, yükseklik, genişlik, num_classes].Tamamen kıvrımlı ağ için piksel softmax değeri

Sorum şu: Tensorflow'da hangi işlev, diğer piksellerden bağımsız olarak her piksel için softmax işlemini uygulayabilir. Tf.nn.softmax ops böyle bir amaç için görünmüyor.

Böyle bir op yoksa, sanırım kendim yazmam gerek.

Teşekkürler!

GÜNCELLEME: Kendimi uygulamak zorunda yoksa, ben giriş tensörünü yeniden şekillendirmek gerekebilir düşünüyorum [N, num_claees] N = Toplu x genişlik x yükseklik ve tf.nn.softmax uygulamak, o zaman yeniden şekillendirmek geri. Mantıklı geliyor?

+0

2d matrisine yeniden şekillendirebilir, softmax yapabilir ve sonra yeniden şekillendirebilirsiniz. – Aaron

+0

Aha, yorumlarınızı aynı anda güncelleyin. Ama teşekkürler! –

+0

Hey, sonunda bunu uyguladın mı? Burada sıkışıp kaldım. –

cevap

3

2d'ye yeniden şekillendirin ve ardından yeniden şekillendirdiğinizde, yeniden tahmin ettiğiniz gibi, doğru yaklaşımdır.

1

Bu işlevi kullanabilirsiniz.

GitHub'dan arama yaparak buldum.