Write a Python version of video dataloader

I can give you a reference to the python version of the video dataloader: import torch from torch.utils.data import DataLoader

class VideoDataloader(DataLoader): def init(self, data, batch_size, shuffle=True): self.data = data self.batch_size = batch_size self.shuffle = shuffle

def __iter__(s