python极大极小搜索 python极值

 sift算法中有一步就是求空间极值点

import numpy as np

def getjizhi(inputs,pad=1,space=1):#输入矩阵，求取范围,边界距离
    output=[]
    inputs=np.array(inputs)
    size=inputs.shape
    if len(size) is 1:
        pass
    if len(size) is 2:
        pass
    if len(size) is 3:
        if isinstance(pad,int):
            pad=np.array([pad,pad,pad])
        if isinstance(space,int):
            space=np.array((space,space,space))
        tpos=np.array(size)-np.array(space)
        tpos=np.mgrid[space[0]:tpos[0],space[1]:tpos[1],space[2]:tpos[2]]#产生需要判断的坐标
        tpos=np.array([x.reshape(x.size) for x in tpos]).T#所有需要判断的坐标
        print(tpos)
        for tp in tpos:
            tposf=tp-pad
            tpose=tp+pad+1
            tpmat=inputs[tposf[0]:tpose[0],tposf[1]:tpose[1],tposf[2]:tpose[2]]
            if inputs[tuple(tp)] == tpmat.max():
                print(tpmat.shape,inputs[tuple(tp)],tpmat.max())
                output.append(tp)#保存特征点
    return output