Full image prediction works with batch size 1 - validation still buggy.