Unroll the image once in predict instead of multiple times.