使用demo时，出现下面的错误

[linglangqiuxun]

in fromarray
mode, rawmode = _fromarray_typemap[typekey]
KeyError: ((1, 1, 64), '|u1')

[WesternTrail]

同样的错误

[mhdwhx]

修改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)

[WesternTrail]

感谢，我降低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: ***@***.***>

[TingFinn]

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!!!

[1596892709]

修改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.

[hellohandsome]

感谢，我降低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的结果怎么样我跑的结果很零碎