Skip to content

matcher代码本身就有问题 #40

@wenwwww

Description

@wenwwww

Matcher.py中的set_reference函数

imgs = imgs.flatten(0, 1)  # bs, 3, h, w
img_size = imgs.shape[-1]# bs*3, h, w
...
self.ref_imgs = imgs

这里直接给self.ref_imgs 一个shape为( bs*3, h, w)的tensor,结果predict函数中的

ref_feats, tar_feat = self.extract_img_feats()

调用self.extract_img_feats()函数,而extract_img_feats()函数中

ref_imgs = torch.cat([self.encoder_transform(rimg)[None, ...] for rimg in self.ref_imgs], dim=0)

要求使用transform进行变换,结果很显然变化直接失败,因为shape根本达不到要求。

Metadata

Metadata

Assignees

No one assigned

    Labels

    No labels
    No labels

    Type

    No type

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions