pythonで「複数枚の画像を順番にハフ変換」したいのですが、うまくいきません。（初心者）

Question

pythonで「複数枚の画像に対して順番にハフ変換」を行いたいのですが、うまくいきません。
下記に実行しているプログラムを載せています。

下記より上の部分で動画を読み込み、その動画を複数のフレーム画像にして「screen_code」というファイルに「数字（0から）.png」で保存しています。そして最後にハフ変換した直線を元の画像に上書きして終了したいのですが、実行すると、 for rho,theta in lines[0]:の部分が「TypeError: 'NoneType' object is not subscriptable」というエラーが出てきます。アドバイスお願いします。

for i in range(1,frame_count):
#画像読み込み
img = cv2.imread('screen_code/'+str(i)+'.png')
#グレースケール
gray = cv2.cvtColor(img,cv2.COLOR_BGR2GRAY)
#エッジ検出
edges = cv2.Canny(gray,50,150,apertureSize = 3)
#ハフ変換
lines = cv2.HoughLines(edges,1,np.pi/180,200)
for rho,theta in lines[0]:
a = np.cos(theta)
b = np.sin(theta)
x0 = arho
y0 = brho
x1 = int(x0 + 1000*(-b))
y1 = int(y0 + 1000*(a))
x2 = int(x0 - 1000*(-b))
y2 = int(y0 - 1000*(a))

cv2.line(img,(x1,y1),(x2,y2),(0,0,255),2)
cv2.imwrite('screen_code/'+str(i)+'.png',img)

Comments

[y_waiwai]

このままではコードが意味不明なので、質問を編集して、<code>ボタン、’’’の枠の中にコードを貼り付けてください

Answer 1

cv2.HoughLines() で直線が検出できない場合は、None が返ってくるので、

if lines is None:
    例外処理

が必要かと思います。

Comments

[0530ryo]

その通りでした。ありがとうございます！！

Answer 2

本当にアドバイスですが、

• lines にはデータは入っていますでしょうか？
• lines にデータが入っていなければ、 edges, np には、データやモジュールが読み込まれているか

を確認してみてください。

おそらく np は numpy モジュールを読み込んで居ると思いますが、
import文が見当たらないため、
「まさか import numpy as np を行頭に書いていないということは無いよなあ」
という心配をしました。

Comments

[0530ryo]

ありがとうございます。参考にさせていただきます。