提取特徵點並定位(缺陷檢測)

項目中需要進行缺陷檢測。先做圖像差分,然後檢測出缺陷並標註缺陷位置。

import cv2
import sys
import numpy as np

# 第一個參數 圖片路徑
imgpath = sys.argv[1]
img = cv2.imread(imgpath)
alg = sys.argv[2]

# 可根據參數選擇算法
def fd(algorithm):
    if algorithm == 'SIFT':
        return cv2.xfeatures2d.SIFT_create()
    if algorithm == 'SURF':
        return cv2.xfeatures2d.SURF_create(float(sys.argv[3]) if
                                           len(sys.argv) == 4 else 4000)

gray = cv2.cvtColor(img, cv2.COLOR_BGR2GRAY)

sift = cv2.xfeatures2d.SIFT_create()
keypoints, descriptor = sift.detectAndCompute(gray, None)


img = cv2.drawKeypoints(image=img, outImage=img, keypoints=keypoints,
                        flags=cv2.DRAW_MATCHES_FLAGS_DRAW_RICH_KEYPOINTS,
                        color=(51, 163, 236))

# 循環獲取點的位置
for i in range(len(keypoints)):
    p = keypoints[i].pt
    print(p)

cv2.imshow('sift', img)
cv2.waitKey()
cv2.destroyAllWindows()

 

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章