Python OpenCV matplotlib.pyplot 画像認識 モザイク エラーの解決方法

Question

下記のコードで出るエラーが解決できません。

import cv2

def mosaic(img,rect,size):
    (x1,y1,x2,y2)=rect
    w=x2-x1
    h=y2-y1
    i_rect=img[y1:y2,x1:x2]
    i_small=cv2.resize(i_rect,(size,size))
    i_big=cv2.resize(i_small,(w,h),interpolation=cv2.INTER_AREA)
    img2=img.copy()
    return img2

import matplotlib.pyplot as plp 

img=cv2.imread("itzy-yunaa.jpg")
mos=mosaic(img,(2585,2585,1885,1885),10)
cv2.imwrite("itzy--yuna.lpg",mos)
plp.imshow(cv2.cvtColor(mos,cv2.COLOR_BGR2RGB))
plt.show()

エラー
Traceback (most recent call last):
File "c:\Users\c7not\OneDrive\デスクトップ\python_lesson\YouTube\secound.py", line 16, in
<module>
mos=mosaic(img,(2585,2585,1885,1885),10)
File "c:\Users\c7not\OneDrive\デスクトップ\python_lesson\YouTube\secound.py", line 8, in mosaic
i_small=cv2.resize(i_rect,(size,size))
cv2.error: OpenCV(4.5.3) C:\Users\runneradmin\AppData\Local\Temp\pip-req-build-1i5nllza\opencv\modules\imgproc\src\resize.cpp:4051: error: (-215:Assertion failed) !ssize.empty() in function 'cv::resize'

Answer 1

２か所ほど誤りがあるようです。

■ １つ目

mosaic 関数の第２引数で rect を渡していますが、パラメータの指定方法が誤っているようです。
rect は (x1,y1,x2,y2) を渡す必要があります。x1,y1が開始座標、x2,y2が終了座標である為、サイズの指定に誤りがあると考えられます。

mos=mosaic(img,(2585,2585,1885,1885),10)

■ ２つ目
下記の強調した部分の処理が抜けているようです。

import cv2
import matplotlib.pyplot as plp

def mosaic(img,rect,size):
    (x1,y1,x2,y2)=rect
    w=x2-x1
    h=y2-y1
    i_rect=img[y1:y2,x1:x2]
    i_small=cv2.resize(i_rect,(size,size))
    i_big=cv2.resize(i_small,(w,h),interpolation=cv2.INTER_AREA)
    img2=img.copy()
    #################################
    img2[y1:y2, x1:x2] = i_big
    #################################
    return img2

img=cv2.imread("duck.jpg")
# (100,100,300,200) … x1,y1,x2,y2
mos=mosaic(img,(100,100,300,200),10)
cv2.imwrite("result.jpg",mos)
plp.imshow(cv2.cvtColor(mos,cv2.COLOR_BGR2RGB))
plp.show()

■ 実行結果
赤枠の部分にモザイクが入っていることが確認できます。（赤枠自体はPythonではなく手で追加したものです。）

＜参考情報＞
下記サイトにモザイク処理のサンプルがありました。
処理にコメントが入っているので、こちらを参照すると理解が深まるのでは、と思います。

■ 【Python】画像にモザイクをかける
https://ymgsapo.com/2018/11/01/mosaic-image/

Comments

[lovely]

ご回答ありがとうございます。
二つ目のコードはどういう意味でしょうか？？

[cx20]

> img2[y1:y2, x1:x2] = i_big

i_big は、モザイク処理をした画像に相当します。
img2[y1:y2, x1:x2] は、赤枠に相当するエリアになります。
モザイク結果を、赤枠の部分に貼り付けているイメージになります。

> img=cv2.imread("duck.jpg")
> # (100,100,300,200) … x1,y1,x2,y2
> mos=mosaic(img,(100,100,300,200),10)
> cv2.imwrite("result.jpg",mos)

ここのコードは、ローカルでテストする際に使った画像データに合わせて、パラメータ調整をしたものになります。
テストに使用したアヒルの画像は512x512の画像データで、モザイク対象は赤枠の部分を (100,100,300,200) とすることで、
(100,100)～(300,200) の範囲にモザイクをかけています。

詳細は下記サイトを参照ください。同様のサンプルが記載されています。

■ 【Python】画像にモザイクをかける
https://ymgsapo.com/2018/11/01/mosaic-image/

> # 元の画像へモザイク画像をコピー
> img2 = img.copy()
> img2[y1:y2, x1:x2] = zoom

[lovely]

ありがとうございます。
どうして渡すときに[x1,y1,x2,y2]の形ではなく：を使っている形なのですか？

[cx20]

> どうして渡すときに[x1,y1,x2,y2]の形ではなく：を使っている形なのですか？

Python の構文にそれほど詳しい分けではありませんが「:」は「スライス」と呼ばれる書き方のようです。
配列に対して「始点のインデックス番号 : 終点のインデックス番号」といった形でアクセスすることが出来ます。
指定範囲の配列にアクセスする為にそのような記述になっているようです。

■ Pythonのスライス
https://qiita.com/kaeruair/items/e7f1c08915839ce3c9b4

img2 は ndarray と呼ばれる N次元配列になっており、今回のケースでは３次元配列 [x][y][rgb] になっています。

以下は、 試しに img2 の内容を print してみた結果です。2x2の4ピクセルの範囲を出力しています。
```python
print(type(img2))
print(img2[100:102, 100:102])
```

指定範囲の色情報が取得できることが確認できます。
```
<class 'numpy.ndarray'>
[[[26 32 27]
[26 32 27]]

[[26 32 27]
[26 32 27]]]
```
[26 32 27] は 青「26」、緑「32」、赤「26」の意味になります。
4つのピクセルとも同じ色なのは、モザイク処理を行って色が同じになった為と考えられます。

なお [x1,y1,x2,y2] の場合、3次元データに4次元データとしてアクセスすることになりエラーになります。
[x1,y1,0] とすると単一のピクセルの色情報が取得できます。座標「x1,y1」の「青」の色情報できます。