前提・実現したいこと
pythonでパワースペクトル画像の特徴を分析したいと思っています.
現在、画像上の角度方向のスペクトル分布を求めてるもの(コード1)とパワースペクトルの角度方向(コード2)を求めるためのものはあります。```
現在、コード1において、読み込んだ画像に2次元FFTをし、そこからパワースペクトルを求めることはできています。ただ、コード1で読み込んだ画像を使ったものをコード2でも適用させる方法がいくら調べてもわかりません。詳しい方に聞くと、mainからdefを呼び出して、returnされた値をplotするだけで、def以下のコードは何も修正する必要はないとのことでした。ただ、知識が浅く、具体的な方法がわかりません。もし、時間があれば回答お願いします。
-- coding: utf-8 --#コード1
import numpy as np
import cv2
import matplotlib.pyplot as plt
def main():
# 入力画像をグレースケールで読み込み
img = cv2.imread("input.png")
# グレースケール変換 gray = cv2.cvtColor(img, cv2.COLOR_BGR2GRAY) # 高速フーリエ変換(2次元) fimg = np.fft.fft2(gray) # 第1象限と第3象限, 第2象限と第4象限を入れ替え fimg = np.fft.fftshift(fimg) # パワースペクトルの計算 mag = 20*np.log(np.abs(fimg)) # 入力画像とスペクトル画像をグラフ描画 plt.subplot(121) plt.imshow(gray, cmap = 'gray') plt.subplot(122) plt.imshow(mag, cmap = 'gray') plt.show()
if name == "main":
main()
#コード2
import numpy as np
複素平面を構成する値の配列を生成(原点は中心)
def complex_plane(width):
half_width = width // 2
re = np.array(range(width)) - half_width
im = - re
re, im = np.meshgrid(re, im)
return re + im * 1j
スペクトルを与えられた角度の範囲内だけ集計
def aggregate_in_angle(agg_fun, spectrum, min_angle, max_angle):
width = spectrum.shape[0]
assert spectrum.shape[1] == width
min_radius, max_radius = 0, width // 2
assert 0 <= min_angle < max_angle <= 180 # 単位はdegree
cp = complex_plane(width) cp_mag = np.abs(cp) in_radius = np.logical_and(min_radius < cp_mag, cp_mag < max_radius) cp_angle = np.angle(cp, deg=True) in_angle = np.logical_and(min_angle <= cp_angle, cp_angle < max_angle) return agg_fun(spectrum[np.logical_and(in_radius, in_angle)])
def test(spectrum, angle_gap):
# 指定角度ごとの平均を求める
means = [aggregate_in_angle(np.mean, spectrum, angle, angle + angle_gap)
for angle in range(0, 180, angle_gap)]
return means
### 発生している問題・エラーメッセージ pythonを始めたばかりで、defについて勉強しているのですが、まだまだわからないことも多く、複数の関数をつなげることがまだできません。
エラーメッセージ
### 該当のソースコード ```ここに言語名を入力python ソースコード
試したこと
ここに問題に対して試したことを記載してください。
補足情報(FW/ツールのバージョンなど)
ここにより詳細な情報を記載してください。
あなたの回答
tips
プレビュー