pytorch 数据加载性能对比

传统方式需要10s，dat方式需要0.6s

import os

import time
import torch
import random
from common.coco_dataset import COCODataset

def gen_data(batch_size,data_path,target_path):
    os.makedirs(target_path,exist_ok=True)
    dataloader = torch.utils.data.DataLoader(COCODataset(data_path,
                                                         (352, 352),
                                                         is_training=False, is_scene=True),
                                             batch_size=batch_size,
                                             shuffle=False, num_workers=0, pin_memory=False,
                                             drop_last=True)  # DataLoader
    start = time.time()
    for step, samples in enumerate(dataloader):
        images, labels, image_paths = samples["image"], samples["label"], samp

文章来源: blog.csdn.net，作者：网奇，版权归原作者所有，如需转载，请联系作者。

原文链接：blog.csdn.net/jacke121/article/details/85236561