(966, 336, 4) (966, 336, 4) 5 [0.42296719 0.40235877 0.3972783  0.51887385]
