import numpy as np
import paddle
import cv2
from paddle.io import DataLoader
from paddle.io import Dataset
import torch.utils.data as data
import torch

class testdata1(data.Dataset):
    def __init__(self):
        super(testdata1,self).__init__()
        self.img_list = [1,2,3,4,5,6,7]
        self.img = [np.ones((45,45,3)),np.ones((60,60,3)),np.ones((60,60,3)),
                    np.ones((60,60,3)),np.ones((60,60,3)),np.ones((60,60,3)),np.ones((60,60,3))]
        self.size = 1200
        self.shapesize = [200,2400]
    def get_rand(self,):
        self.size = np.random.choice(self.shapesize,1)[0]
    def __len__(self):
        return len(self.img_list)
    def __getitem__(self, index):
        x = cv2.resize(self.img[index],(self.size,self.size))
        x = torch.Tensor(x).permute((2,0,1))
        return x

data_loader = testdata1()
dataload =data.DataLoader(data_loader,
                batch_size=2,
                shuffle=True,
                num_workers=0,
                drop_last=True)

for _,x in enumerate(dataload):
    data_loader.get_rand()
    print(x.shape)

class testdata(Dataset):
    def __init__(self):
        super(testdata,self).__init__()
        self.img_list = [1,2,3,4,5,6,7]
        self.img = [np.ones((45,45,3)),np.ones((60,60,3)),np.ones((60,60,3)),
                    np.ones((60,60,3)),np.ones((60,60,3)),np.ones((60,60,3)),np.ones((60,60,3))]
        self.size = 1200
        self.shapesize = [200,2400]
    def get_rand(self,):
        self.size = np.random.choice(self.shapesize,1)[0]
    def __len__(self):
        return len(self.img_list)
    def __getitem__(self, index):
        x = cv2.resize(self.img[index],(self.size,self.size))
        x = paddle.to_tensor(x).transpose((2,0,1))
        return x

data_loader = testdata()
dataload =DataLoader(data_loader,
                batch_size=2,
                shuffle=True,
                num_workers=0,
                drop_last=True)

for _,x in enumerate(dataload):
    data_loader.get_rand()
    print(x.shape)

这里提供了torch的数据加载和paddle2.1.2加载的简单示例，torch可以修改，但是paddle会报错，