Python скрипт для MTF - PullRequest
Новая
       6

Python скрипт для MTF

1 голос
/ 10 апреля 2020

Итак, я хочу реализовать скрипт MTF (функция передачи модуляции) в python. В этот скрипт я хочу передать изображение в качестве аргумента и получить значение MTF в качестве результата. это значение MTF будет использоваться для количественной оценки резкости изображения в проекте. Вот некоторые ссылки, которые я использовал в качестве ссылки для разработки сценария - [в настоящее время я использую эту ссылку для разработки сценария] https://github.com/weiliu4/py_mtf/blob/master/mtf.py https://github.com/habi/GlobalDiagnostiX/blob/master/MTF.py

Определение MTF - MTF определяется как FFT функции разброса строк. Функция разброса линий определяется как производная от функции разброса границ. Функция разбрасывания кромки - это значения вдоль кромки, в идеале - цель теста с острием. 

def ESF(path):
    img=cv2.imread(path)
    img=cv2.cvtColor(img, cv2.COLOR_BGR2GRAY)
    #m_out=0.299*img[:,:,0]+0.589*img[:,:,1]+0.114*img[:,:,2]
    #x=m_out[100,:]
    x=img
    mu=np.sum(x)/(x.shape[0])
    tmp=(x[:]-mu)**2
    sigma=np.sqrt(np.sum(tmp)/(x.shape[0]))
    edge_function=(x[:]-mu)/sigma
    #print(edge_function)
    edge_function=edge_function[::3]
    #lsf=edge_function[:-2]-edge_function[2:]
    lsf=np.abs(np.diff(edge_function))
    mtf=abs(np.fft.fft(lsf))
    print(mtf)
    #print(np.max(mtf))
    mtf=mtf[:]/np.max(mtf)
    mtf=mtf[:len(mtf)//2]
    mtf=np.arange(mtf.shape[0]*mtf.shape[1]).reshape(mtf.shape[0],mtf.shape[1])
    ix=np.arange(mtf.shape[0])/(mtf.shape[0])
    mtf_poly=np.polyfit(ix, mtf,6)
    #print(mtf_poly)
    #mtf_poly=np.squeeze(mtf_poly)
    #print(mtf.shape[0])
    #print(mtf.shape[1])
    #poly=np.poly1d(mtf_poly)
    #print(poly)
    plt.figure()
    plt.title("MTF")
    plt.xlabel(r'Frecuency $[cycles/pixel]$') ; plt.ylabel('mtf')
    p= plt.plot(mtf,'-or')
    #ll = plt.plot(poly(ix))
    #plt.legend([p,ll],["MTF values","polynomial fit"])
    plt.grid()
    #plt.show()

currently I am getting error while evaluating the line

poly = np.poly1d (mtf_poly)

Любая помощь в том, что я я делаю неправильно Также может кто-то, пожалуйста, дайте мне знать, если я направляюсь в правильном направлении, чтобы найти MTF. ТИА

...