Python OpenCV 直線検出のプログラムでループ処理を行った場合の直線の描画

Question

前提・実現したいこと

http://rikoubou.hatenablog.com/entry/2019/04/03/163751
の直線検出プログラムをもとに, 閾値(threshold)をループ処理して最後のループの直線のみを描画したいです.

発生している問題

threshold値35~70全てのループの直線を描画してしまう状態です.
(イメージとしては紙に絵の具で何本も線を書いて, さらにその上から同じ色の絵の具で別の線を描くような感覚です.)

つまり最後のループの直線のみを描画するか, threshold値35で描画した直線をthreshold値36になったときに一度消す方法があれば教えていただけたら幸いです.
(画像ごとに閾値が自動で変わる, つまりループがどの閾値で終わるかわからないというプログラムを想定しているので, 初めから閾値を70にして直線を描画するという方法は無しでお願いします.)

該当のソースコード

import numpy as np
import cv2

IMAGE_PATH = "./test.jpg" # 読み込む画像

Z = 34 # threshold(閾値)の初期値

def main():
    image  = cv2.imread(IMAGE_PATH) # 画像読み込み
    image2 = cv2.imread(IMAGE_PATH) # 画像読み込み
    gray = cv2.cvtColor(image,cv2.COLOR_BGR2GRAY) # グレースケール化
    outLineImage = cv2.Canny(gray, 220, 250, apertureSize = 3)   # 輪郭線抽出

    cv2.imwrite("./outLine.png", outLineImage)    # ファイル保存

    hough  = hough_lines(image, outLineImage)     # ハフ変換による直線抽出
    cv2.imwrite("./result_hough.png", image)    # ファイル保存


# ハフ変換で直線を抽出する関数
def hough_lines(image, outLineImage):
    while True:
        global Z
        Z += 1
　　　　result = image
    　　lines = cv2.HoughLines(outLineImage, rho=1, theta=np.pi/180, threshold=Z) # ハフ変換で直線抽出
    　　print("hough_lines: ", len(lines))

    　　for line in lines:
        　　rho, theta = line[0]
        　　a = np.cos(theta)
        　　b = np.sin(theta)
        　　x0 = a*rho
        　　y0 = b*rho
        　　x1 = int(x0 + 1000*(-b))
        　　y1 = int(y0 + 1000*(a))
        　　x2 = int(x0 - 1000*(-b))
        　　y2 = int(y0 - 1000*(a))

        cv2.line(result,(x1,y1),(x2,y2),(0,0,255),2) # 赤色で直線を引く

　　if threshold < 71
　　　　 break # 閾値が適正値になったので終了
　　
　　lines = np.zeros_like(lines) # linesの初期化

    return result

if __name__ == '__main__':
    main()

補足情報（FW/ツールのバージョンなど）

Python3.7, OpenCV4.1.2

Comments

[meg_]

リンクは「リンクの挿入」で記入してください。

[can110]

提示コードにはインデントに全角空白がふくまれているようです。
また「if threshold < 71」にて「:」がないため文法エラーになります。
第三者が実行できるようにコードを修正ください。

Answer 1

閾値を変えるごとに，線を描画しているバッファを適切に初期化すればよいだけではないでしょうか．
（元絵の上に線を描いたような結果が欲しいならば，最も簡単には，毎度Matをcloneして描画バッファを用意すればよいのでは）

---

あと，質問内容とは無関係ですが，HoughLinesへの入力画像は，
"8-bit, single-channel binary source image. The image may be modified by the function. "
とされているので，そのようなループで処理するのはまずい予感．