python 画像認識

Question

pyautogui を用いて画像認識を行っています。

try:
locate = pyautogui.locateCenterOnScreen('search.png')
pg.click(locate)
except:
pg.alert(text='探索失敗', button='OK')

上記ソースで、pyautogui.locateCenterOnScreen('search.png')が見つからなかった場合に
例外(Except)で補足しています。

上記でも実装できるのですが、IF文のような形で分岐させることはできないでしょうか。

IF pyautogui.locateCenterOnScreen('search.png')
画像発見時の処理
ELSE
画像がなかった時の処理
ENDIF

という風に記述をしたいと考えています。

可能かどうか教えていただきたいです。

Answer 1

関数で処理をラップすれば可能です。

def locateCenterOnScreen(path):
    try:
        pyautogui.locateCenterOnScreen('search.png')
        return True
    except 例外クラス:
        return False

ただし『記述できる』だけであって、実際に用いるべきかはまた別の話です。

Answer 2

pyautogui.locateCenterOnScreen のドキュメントはこちらにあります。

https://pyautogui.readthedocs.io/en/latest/screenshot.html#the-locate-functions

これをみると、見つからない場合は ImageNotFoundException を投げるという仕様になっています。
したがって、例外を用いない方法は不可能です。

コメントを受けての追記: 例外の補足の仕方

ImageNotFoundException は pyscreeze で定義されていますので、次のように import してから使ってください。

from pyscreeze import ImageNotFoundException 

try:
    locate = pg.locateCenterOnScreen('search3.png')
    pg.click(locate)
except ImageNotFoundException:
    pg.alert(text='探索失敗', button='OK')

コメントを受けての追記: pyscreeze の設定変更の仕方

pyautogui の内部で使われている pyscreeze が2度仕様変更しましたが、pyautogui が2度目の仕様変更に追従していないようです。

このため、pyscreeze が2度目の仕様変更の前の振る舞いをするように、以下のように設定してやる必要があります。

import pyautogui as pg
from pyscreeze import ImageNotFoundException 

# pyscreeze の設定
import pyscreeze
pyscreeze.USE_IMAGE_NOT_FOUND_EXCEPTION = True

try:
    locate = pg.locateCenterOnScreen('search3.png')
    pg.click(locate)
except ImageNotFoundException:
    pg.alert(text='探索失敗', button='OK')

Comments

[Wish_N]

回答いただきありがとうございます。

try:
locate = pg.locateCenterOnScreen('search3.png')
pg.click(locate)
except ImageNotFoundException:
pg.alert(text='探索失敗', button='OK')

上記のように例外を補足した場合、
name 'ImageNotFoundException' is not defined
と出てしまいます。　例外の捕捉方法が誤っていますでしょうか。

[nunukim]

回答に追記しました

[Wish_N]

nunukimさん

回答ありがとうございます。
下記のように修正したところ、NoneType' object is not subscriptable　という例外が発生してしまいました。

import subprocess
import pyautogui as pg
from pyscreeze import ImageNotFoundException

try:
locate = pg.locateCenterOnScreen('search3.png')
pg.click(locate)
except ImageNotFoundException:
pg.alert(text='探索失敗', button='OK')

Exceptで補足出来ないところで落ちなくはなったのですが、これの原因はわかりますでしょうか。

[nunukim]

pyscreeze に以下のような記述がありました。

# For version 0.1.19 I changed it so that ImageNotFoundException was raised
# instead of returning None. In hindsight, this change came too late, so I'm
# changing it back to returning None. But I'm also including this option for
# folks who would rather have it raise an exception.

pyautogui は pyscreeze を使っていますが、pyscreeze が2度仕様変更した(一度変更したものを元に戻した)のに対し、
pyautogui は2回目の仕様変更に追従していないようです。

このため、pyscreeze が2度目の仕様変更の前の振る舞いをするように設定を変えてやる必要があります。
設定のしかたを回答に追記します。

[nunukim]

追記しました。

[Wish_N]

nunukimさん
回答ありがとうございます。
ご丁寧にありがとうございます。　無事実装出来ました。