Tensorfow.js tensörün medyanını bulur

Question

Açısal uygulamada tensorflow.js aracılığıyla tensör olarak okunan bir görüntünün medyanını bulmam gerekiyor. Herhangi biri tfj'lerde medyanın nasıl bulunacağını önerebilir mi?

Sevmek -

let x = tf.tensor([1, 2, 3, 4, 5, 8, 9]);
console.log(x.median());

Bu, 4 yazdırmalıdır.

Javascript'te tensorflow olasılık istatistiklerinin eşdeğerini bulmaya çalıştım ama şimdiye kadar şansım olmadı.

Answer 1

Ortanca, seri düzenlendikten sonraki ortadaki sayıdır. İlk önce tensörün sıralanması gerekir. Endeksinde değer sizeTensor / 2medyan değeri olacaktır.

t= tf.tensor([1, 2, 3, 4, 5, 8, 9]);
tf.topk(t, t.size).values.slice(t.size / 2, 1).print() // will print 4

Yukarıdakilerin yüksek boyutlu tensörlerle çalışması için - örneğin görüntüler, önce 1d tensöre yeniden şekillendirmemiz gerekecek

imgBir görüntünün tensörü göz önüne alındığında

t = img.reshape([-1])
// do the same as above