import torch


def collate_fn_customer(batch):
    """
    这个函数的作用是将读取到的batch中的多组数据,融合成整体
    也就是增加一个batch维度
    """
    images = []
    bboxes = []
    for i, data in enumerate(batch):
        # data[0]为img维度
        images.append(data[0])
        # data[1]为bbox维度
        bboxes.append(data[1])
    
    #images类型转换:list==>torch.tensor
    images = torch.stack(images)
    batch = (images, bboxes)
    return batch