ある値を順番に代入し、代入した結果が最大となる時の値を求めたいです。

Question

以下のコードでは、読み込んだ画像の座標(4692,2955)を中心とする半径900の円を描き、その円周上の輝度の合計を求めて出力しています。
この半径の値を800から1000まで順番に変化させ、輝度の合計が最大となる半径rを求めたいです。

import cv2
import numpy as np

img = cv2.imread('circle.png')

img_mask = np.zeros(img.shape[:2], dtype=np.uint8)
cv2.circle( img_mask,
            (4692,2955), #円の中心座標
            900, #円の半径　<-ここに800~1000まで順番に代入したい
            1,
            thickness=30,
            lineType=cv2.LINE_4,
            shift=0)

img = cv2.cvtColor(img, cv2.COLOR_BGR2GRAY)
ret = np.sum(img, where=img_mask == 1)
print(ret)

半径rをr=list(range(800,1001))
としてfor文を用いて順番に入れていけば良いのかなと思うのですが、具体的なコードの書き方がわからず困っています。

詳しい方お力をお借りしたいです。

Comments

[fana]

質問内容と直接関係ない話ですが……
np.sum というやつがどれだけ高速なのか知りませんが，それを 201回繰り返すというのは処理速度的に問題ないのかな…？　とか，ちょっとだけ思いました．

例えば，適当なバッファ（配列みたいなの）を用意して，
各画素が自身の輝度値を，そのバッファの「円の中心からその画素までの距離」に相当する場所に加算していくような処理を考えれば，画像走査自体は１回で済むように思いますが…

Answer 1

現在のコードを半径を引数として受け取り、輝度の合計を返す関数に書き換えましょう。

半径rをr=list(range(800,1001)としてfor文を用いて、その中で定義した関数を呼び出し、最大値を与えるときの半径を取り出せばよいでしょう。

Comments

[mappys]

ppaul様　回答誠にありがとうございます。

以下自分なりにコードを書いてみました。最大値を呼び出すところままだできていないのですが、輝度の合計を返す関数を定義して半径rのリストを定義した関数に入れていく、ということをしたいのですが、、、

def内の半径のパラメータにrを使っているのがよくない気がしています、、、

import cv2
import numpy as np

img = cv2.imread('circle.png')

r_list = list(range(800,1001))

#defで半径rの円周上の輝度の合計を求める関数を定義したい
def func(r):
img_mask = np.zeros(img.shape[:2], dtype=np.uint8)
cv2.circle( img_mask,
(4692,2955), #円の中心座標
r,　<-ここでrを使って良いのかわかりません
1,
thickness=30,
lineType=cv2.LINE_4,
shift=0)
img = cv2.cvtColor(img, cv2.COLOR_BGR2GRAY)
ret = np.sum(img, where=img_mask == 1)
return ret

for r in r_list:
print(func(r))

エラーコードは以下となりました
Traceback (most recent call last):
File "/Users/masashi/pyworks/circle_for.py", line 23, in <module>
print(func(r))
File "/Users/masashi/pyworks/circle_for.py", line 10, in func
img_mask = np.zeros(img.shape[:2], dtype=np.uint8)
UnboundLocalError: local variable 'img' referenced before assignment

[ppaul]

imgという変数の使いかたの問題です。
img = cv2.cvtColor(img, cv2.COLOR_BGR2GRAY) でimgを定義しています。
そしてimgのglobal宣言はありません。このためimgはローカル変数であると判断されています。
その結果、img_mask = np.zeros(img.shape[:2], dtype=np.uint8) でimgを参照しているのが定義前参照となります。
https://docs.python.org/ja/3/reference/executionmodel.html#naming-and-binding
にその説明があるのですが、わかりにくいかもしれません。

[mappys]

ありがとうございます、引用していただいたサイトをもとに色々試してみます。