在复现一篇论文的时候,碰到需要将原图片放大一倍的需求,由于精度要求不高,网上给出的方法太过繁琐,就自己写一个简易的算法来实现。
输入的是一个黑白图片矩阵。主要的思路是在两个像素点之间进行插值,如果两个像素点都是0,插值0,有一个为1,插值1,这么做会稍微扩大1的范围,但基本还是保持了原来的形状,代码如下:
def augment_pic(pic):#pic:2d array
aug_pic_line = []
aug_pic = []
for i in range(len(pic)):
line = []
for k in range(len(pic[i])-1):
if pic[i][k] == 1 or pic[i][k+1] == 1:
line.extend([pic[i][k], 1])
else:
line.extend([pic[i][k], 0])
line.append(pic[i][-1])
aug_pic_line.append(line)
aug_pic_line_T = np.array(aug_pic_line).T
for i in range(len(aug_pic_line_T)):
line = []
for k in range(len(aug_pic_line_T[i])-1):
if aug_pic_line_T[i][k] == 1 or aug_pic_line_T[i][k+1] == 1:
line.extend([aug_pic_line_T[i][k], 1])
else:
line.extend([aug_pic_line_T[i][k], 0])
line.append(aug_pic_line_T[i][-1])
aug_pic.append(line)
aug_pic = np.array(aug_pic).T
return aug_pic
放大前后的图片展示如下:
放大前图像:
放大后图像: