imsitu_model.py 文件源码

def __call__(self, imgs):
        img_feats = self.resnet152(imgs)

        if not self.is_zeroshot:
            return img_feats

        att_res = self.att_linear(img_feats) if self.att_dim is not None else None
        embed_res = self.embed_linear(img_feats) if self.embed_dim is not None else None
        return ZSResult(att_res, embed_res)