tgif.py 文件源码

def __init__(self, data, caps_field, is_train=True, cnn_size=224, scale_size=256):
        """
        :param all_sents: List of (input, target) sentences
        :param vocab: Vocab
        :param is_train:
        :param num_test:
        """
        self.data = data
        self.caps_field = caps_field
        self.is_train = is_train

        crop = RandomCrop if self.is_train else CenterCrop
        self.transform = transforms.Compose([
            Scale(scale_size),
            crop(cnn_size),
            ToTensor(),
            normalize,
        ])