STANet
STANet copied to clipboard
使用demo时,出现下面的错误
in fromarray mode, rawmode = _fromarray_typemap[typekey] KeyError: ((1, 1, 64), '|u1')
同样的错误
修改CDFA_model中代码 self.dist = F.pairwise_distance(self.feat_A, self.feat_B, keepdim=True) # 特征距离 为 self.feat_A = self.feat_A.permute(0, 2, 3, 1) self.feat_B = self.feat_B.permute(0, 2, 3, 1) self.dist = F.pairwise_distance(self.feat_A, self.feat_B, keepdim=True) # 特征距离 self.dist = self.dist.permute(0,3, 1, 2)
感谢,我降低torch版本可行。发自我的华为手机-------- 原始邮件 --------发件人: mhdkssygit @.>日期: 2023年3月17日周五 晚上7:42收件人: justchenhao/STANet @.>抄送: 陈铭 @.>, Comment @.>主 题: Re: [justchenhao/STANet] 使用demo时,出现下面的错误 (Issue #99) 修改CDFA_model中代码 self.dist = F.pairwise_distance(self.feat_A, self.feat_B, keepdim=True) # 特征距离 为 self.feat_A = self.feat_A.permute(0, 2, 3, 1) self.feat_B = self.feat_B.permute(0, 2, 3, 1) self.dist = F.pairwise_distance(self.feat_A, self.feat_B, keepdim=True) # 特征距离 self.dist = self.dist.permute(0,3, 1, 2)
—Reply to this email directly, view it on GitHub, or unsubscribe.You are receiving this because you commented.Message ID: @.***>
self.feat_A = self.feat_A.permute(0, 2, 3, 1) self.feat_B = self.feat_B.permute(0, 2, 3, 1) self.dist = F.pairwise_distance(self.feat_A, self.feat_B, keepdim=True) # 特征距离 self.dist = self.dist.permute(0,3, 1, 2)
That is very useful for me. Thank you very much!!!
修改CDFA_model中代码 self.dist = F.pairwise_distance(self.feat_A, self.feat_B, keepdim=True) # 特征距离 为 self.feat_A = self.feat_A.permute(0, 2, 3, 1) self.feat_B = self.feat_B.permute(0, 2, 3, 1) self.dist = F.pairwise_distance(self.feat_A, self.feat_B, keepdim=True) # 特征距离 self.dist = self.dist.permute(0,3, 1, 2)
It works, thanks.
感谢,我降低torch版本可行。发自我的华为手机-------- 原始邮件 --------发件人: mhdkssygit @.>日期: 2023年3月17日周五 晚上7:42收件人: justchenhao/STANet _@**._>抄送: 陈铭 _@.>, Comment @._>主 题: Re: [justchenhao/STANet] 使用demo时,出现下面的错误 (Issue #99) 修改CDFA_model中代码 self.dist = F.pairwise_distance(self.feat_A, self.feat_B, keepdim=True) # 特征距离 为 self.feat_A = self.feat_A.permute(0, 2, 3, 1) self.feat_B = self.feat_B.permute(0, 2, 3, 1) self.dist = F.pairwise_distance(self.feat_A, self.feat_B, keepdim=True) # 特征距离 self.dist = self.dist.permute(0,3, 1, 2) —Reply to this email directly, view it on GitHub, or unsubscribe.You are receiving this because you commented.Message ID: _@_.*>
请问一下你用预训练权重跑的 demo的结果怎么样我跑的结果很零碎