使用numba加速图像处理速度

前几天有个任务是需要对图片遍历像素处理，先用python写了一遍。结果由于图像像素质量很高，运行速度很慢。想到既然是像素操作能不能用gpu来分担cpu的工作以加速运行。结果发现了这个好东西：numba。国内对这个库的使用分享并不多，所以留个印记，帮助有同样需求的人。

1.效果

下面两份代码是一样的效果。读取文件夹下面的每张图片，把每张图片灰度化后，灰度值为38的改成1，75的改成2，其余改成0(独热处理label：具体使用场景见https://blog.csdn.net/liu506039293/article/details/100587498)。使用numba加速后的代码速度大概是不使用的20倍。也就是本来需要跑6个小时的代码，现在只要20分钟就搞定了。真是相当好用的工具！！

1.1采用numba加速

import cv2
import os
import numpy as np
from numba import jit
def get_all_files(bg_path):
    files = []

    for f in os.listdir(bg_path):
        if os.path.isfile(os.path.join(bg_path, f)):
            files.append(os.path.join(bg_path, f))
        else:
            files.extend(get_all_files(os.path.join(bg_path, f)))
    files.sort(key=lambda x: int(x[-6:-4]))#排序从小到大
    return f