how convert a list to a PrefetchDataset in python?

Question

i am trying to do some image segmentation, but while making the data prepossessing i found a problem; i use this code to load all the images from train and test directory