torch自定义随机加载

 

  import random

    data_img = []
    data_lable = []

    m_index_i = 0
    for astep, samples in enumerate(dataloader):
        start = time.time()
        images, labels = samples["image"], samples["label"]

        for i in range(images.size(0)):
            data_img.append(images[i, :, :, :])
            data_lable.append(labels[i, :, :])

            asamples = torch.Tensor(config["batch_size"] * len(config["parallels"]), 3, config["img_w"],
                                    config["img_h"])
            asamples[0] = data_img[m_index_i]
            m_index_i += 1
    m_index = [i for i in range(len(data_img))]

    # print("len(data_img)",len(data_img))
    # for j in range(200):
    #     random.shuffle(m_index)
    #     timestr = datetime.dateti