Closed ZhangIceNight closed 1 year ago
我这样做了一点简单修改,目前可以正常运行了
if auto_pad and (q_pad_b > 0 or q_pad_r > 0):
if q_pad_b == 0:
output = output[:, :, :, :-q_pad_r]
elif q_pad_r == 0:
output = output[:, :, :-q_pad_b, :]
else:
output = output[:, :, :-q_pad_b, :-q_pad_r]
我这样做了一点简单修改,目前可以正常运行了
if auto_pad and (q_pad_b > 0 or q_pad_r > 0): if q_pad_b == 0: output = output[:, :, :, :-q_pad_r] elif q_pad_r == 0: output = output[:, :, :-q_pad_b, :] else: output = output[:, :, :-q_pad_b, :-q_pad_r]
感谢指出,我会尽快修复这个问题。
已修复
你好,我发现在新版本的nchw版本的代码中,rrsda.py 文件
如果
q_pad_b
和q_pad_r
中只有一个不为零,那下面的表达式 就会出现[:,:,:-0,:-q_pad_r]
这类情况