コーナー座標の取得をしたいのですが、方法がわかりません。

Question

初めまして。 初心者です。

picureName.pngと言う名前のファイルを読み込み、白黒反転したのち、grayスケールに変換、その後
cornerHarris関数を適用することで、検出されたコーナーに赤色のドットを付記するプログラムを書きました。

image = cv2.imread("pictureName.png")
image = 255 - image
gray = cv2.cvtColor(image,cv2.COLOR_BGR2GRAY)
gray= np.float32(gray)
dst = cv2.cornerHarris(gray,2,9,0.16)
dst = cv2.dilate(dst, None)
image[dst>0.01*dst.max()]= [0,0,255]
cv2.imshow('imageWithCorner',image)

このイメージ上に付記された赤い点の座標を網羅的に取り出したいのですが、どのよな命令文を書けばよいでしょうか？

なお、画像をfloat型に変換していますが、赤い点の座標はfloat型ではなくても構いません。

宜しくお願い致します。

Answer 1

これでいかがでしょうか？

# -*- coding: utf-8 -*-
import cv2
import numpy as np
import sys

image = cv2.imread("pictureName.png", 1)

if image is None:
    # If an image was not loaded, it may cause tuple error.
    sys.exit("File not found.")

image = 255 - image
gray = cv2.cvtColor(image,cv2.COLOR_BGR2GRAY)
gray= np.float32(gray)

# Detect corner
dst = cv2.cornerHarris(gray,2,9,0.16)

# Expand corner
# dst = cv2.dilate(dst, None) 

# Draw red points
image[dst>0.01*dst.max()]= [0,0,255]

# Detect red pixel
coord = np.where(np.all(image == (0, 0, 255), axis=-1))

# Print coordinate
for i in range(len(coord[0])):
    print("X:%s Y:%s"%(coord[1][i],coord[0][i]))

cv2.imshow('imageWithCorner',image)
cv2.waitKey(0)

• XY座標がcoordの中で反転しているので[1]、[0]の順番にしています。そこだけ注意ください。
• cv2.dilateをするとターゲットの大きさが1pxより広がり「コーナー判定箇所」が増えすぎるのでコメントアウトしました

Comments

[mich1111]

slash様
早速ありがとうございます。完璧に動きました。
非常に綺麗なコードで解説も丁寧で助かりました。
御礼申し上げます。