Can you share the code for visualization and prediction

[crzjustlearn]

At present, there is only training code. I want to see the visual results and the judgment of images using the network