Right now, most of the examples just give out image, lables based upon the folder sturcture. How do i write a dataloader/dataset, such that i can use it for a network that take input image and outputs image?