Hy @hfdp, what I understood from your above explanation is that how can you use your trained model for a single image.
Considering a trained model and a image.
# Just set the batch size to 1.
#The model expects batch size to be inserted.
image = image.reshape(1,channels,height,width)
output = model(image)

Hope this helps