最近的对图像数据进行处理的时候需要将图像中的某个颜色替换为另一个颜色,但是网络上找到的方法都是通过对图像的遍历进行替换,实在是太费时了!刚开始使用时觉得CPU很快了,一张图片应该用不了多久,但是实际使用中耗时确实难以接受的!于是自己写了一个替换程序加快速度,比遍历快很多,但我觉得不是最快的,应该有通过矩阵索引更快的处理方式,只是我自己暂时并不知道该如何实现,如果以后能够实现会进行更新,暂时先写下自己暂时觉得可用的代码。
一、通过遍历替换
将图像中某个颜色替换为另一个颜色一般的做法是遍历整个图像,逐一替换,如下:
def replace_color_tran(img, src_clr, dst_clr):
''' 通过遍历颜色替换程序
@paramimg:图像矩阵
@paramsrc_clr:需要替换的颜色(r,g,b)
@paramdst_clr:目标颜色(r,g,b)
@return替换后的图像矩阵
'''
img_arr = np.asarray(img, dtype=np.double)
dst_arr = img_arr.copy()
for i in range(img_arr.shape[1]):
for j in range(img_arr.shape[0]):
if (img_arr[j][i] == src_clr)[0] == True:
dst_arr[j][i] = dst_clr
return np.asarray(dst_arr, dtype=np.uint8)
二、通过矩阵操作加快替换
但是这样做,处理速度是很慢的即便是现在CPU很快的情况下。我自己通过numpy矩阵操作将速度提升了一点,具体做法如下:
将图像的三个通道拆分开来为R,G,B三个通道
将三个通道的数据值进行简单的编码,合并为单通道矩阵;
将需要替换的颜色进行同2的编码