(128, 2, 224, 224, 3)
image a shape
(128, 224, 224, 3)
labels
tf.Tensor(2, shape=(), dtype=int32)
prediction a
tf.Tensor([ 0.15290104 -0.11335965  0.05952606  0.63431394 -0.6388165 ], shape=(5,), dtype=float32)
(128, 2, 224, 224, 3)
image a shape
(128, 224, 224, 3)
labels
tf.Tensor(0, shape=(), dtype=int32)
prediction a
tf.Tensor([ 1.2271674   0.06227781 -0.06310828  0.509202    0.16954662], shape=(5,), dtype=float32)
(128, 2, 224, 224, 3)
image a shape
(128, 224, 224, 3)
labels
tf.Tensor(4, shape=(), dtype=int32)
prediction a
tf.Tensor([-0.34474376  0.09270601 -0.44066426 -0.08329549 -0.04816262], shape=(5,), dtype=float32)
(128, 2, 224, 224, 3)
image a shape
(128, 224, 224, 3)
labels
tf.Tensor(0, shape=(), dtype=int32)
prediction a
tf.Tensor([ 0.8306643   0.03662108 -0.40713277 -0.44725904 -0.27027306], shape=(5,), dtype=float32)
(128, 2, 224, 224, 3)
image a shape
(128, 224, 224, 3)
labels
tf.Tensor(3, shape=(), dtype=int32)
prediction a
tf.Tensor([ 0.6583673  -0.3735458   0.00993944  1.0526401   0.26237836], shape=(5,), dtype=float32)