我正在尝试手动进行零填充,但出现以下错误:
ValueError: 操作数无法与形状 (2,) (512,3) 一起广播
如果有人可以在不使用内置函数的情况下更正此代码或提供手册。谢谢。
def pad(img,layers):
#img should be rectangular
return [[0]*(len(img[0])+2*layers)]*layers + \
[[0]*layers+r+[0]*layers for r in img] + \
[[0]*(len(img[0])+2*layers)]*layers
pad(img,2)