-
Notifications
You must be signed in to change notification settings - Fork 40
Open
Description
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
Labels
No labels