import cv2
import numpy as np

SOURCE_IMAGE='../lisbon1.jpg'
OUTPUT_IMAGE='orb_lisbon1.jpg'

## kép beolvasása
img = cv2.imread(SOURCE_IMAGE);

## a képet szürkeárnyalatossá konvertáljuk
gray_img = cv2.cvtColor(img, cv2.COLOR_BGR2GRAY)

## jellemzőpontok detektálása
nFeaturePoints = 1000
patchSize = 10
scroreType = cv2.ORB_FAST_SCORE
orb = cv2.ORB_create(nfeatures=nFeaturePoints, scoreType=scroreType, patchSize=patchSize)
keypoints = orb.detect(gray_img)

## kulcspont leírók számítása
keypoints, descriptors = orb.compute(gray_img, keypoints)

## kulcspontok kirajzolása
out_img = cv2.drawKeypoints(gray_img, keypoints, descriptors, color=(255, 0, 255), flags=cv2.DRAW_MATCHES_FLAGS_DRAW_RICH_KEYPOINTS)

cv2.imwrite(OUTPUT_IMAGE, out_img)