You can use this to get a batch of one image:
def repeat_one(source, n=64):
"""Single image helper for displaying batch"""
return [get_image_files(source)[9]]*n
If you want to play around with image augmentations you could also look at this