前提
画像処理の勉強をしていて、以下のサイトで書かれているRichardson_Lucy法を勉強しています。
この中の一次元のものを実装しようとしているのですが、うまく行きません。
https://qiita.com/yusuke_s_yusuke/items/c0e3ba1cf26e96e6696b#rl法のシミュレーション
発生している問題・エラーメッセージ
Traceback (most recent call last): File "test2_simulation.py", line 34, in <module> rl_img = restoration.richardson_lucy(obtained_img, psf, iterations) File "/Users/.pyenv/versions/anaconda3-5.3.1/lib/python3.7/site-packages/skimage/restoration/deconvolution.py", line 367, in richardson_lucy psf_mirror = psf[::-1, ::-1] IndexError: too many indices for array: array is 1-dimensional, but 2 were indexed
該当のソースコード
以下の、
rl_img = restoration.richardson_lucy(obtained_img, psf, iterations)
というところでエラーが出てしまいます。
python3
1import numpy as np 2import matplotlib.pyplot as plt 3import matplotlib.colors as colors 4import matplotlib.cm as cm 5from scipy.stats import norm # おまけの項で使います 6from scipy.stats import multivariate_normal 7from scipy.signal import convolve 8from skimage import restoration 9 10 11"""シミュレーション(1次元バージョン)""" 12# 真の画像 13true_px = 128 14true_img = np.zeros(true_px) 15true_img[52] = 0.5 16true_img[76] = 0.5 17 18# PSF 19psf_px = 127 20mean = psf_px // 2 21psf_sigma = 13 22psf = norm.pdf(np.arange(psf_px), loc=mean, scale=psf_sigma) 23 24 25# 真の画像をPSFで畳み込む(撮像画像) 26obtained_img = convolve(true_img, psf, "same") 27 28plt.plot(psf) 29plt.plot(obtained_img) 30plt.show() 31 32# RL法 33iterations = 50 34rl_img = restoration.richardson_lucy(obtained_img, psf, iterations)
scikit-imageについてのHPは以下のサイトです。
https://scikit-image.org/docs/stable/api/skimage.restoration.html
試したこと
インデックスの問題なのだと思いますが、どこが余計に指定されているのかがわかりません。
何かご存知の方がいらっしゃれば、教えていただきたいです。
補足情報(FW/ツールのバージョンなど)
python 3.7.9
エラーが出た行だけでなく、その前後の関連するコードも貼りましょう。
申し訳ありません。追加修正いたしました。
エラーが出たなら、エラーメッセージを提示しましょう
エラーメッセージは、よけいな省略翻訳しないで出たそのママをコピペで提示してください
申し訳ありません。エラーメッセージも追加いたしました。
そのコードで問題なく動いて、綺麗なグラフが表示されましたけど…。
ちなみに、macOS 付属の python 3.8.9 と Homebrew で入れた 3.9.13 で試しました。ライブラリは最新のを入れました。
エラーが発生している、
psf_mirror = psf[::-1, ::-1]
ですが、scikit_image 0.19.2 では、
psf_mirror = np.flip(psf)
となっています。なので、scikit_image を最新版にアップデートしてみて下さい。
scikit_imageを最新版にしたら解決しました!ありがとうございます!
回答1件
あなたの回答
tips
プレビュー