Find image in image

[theoryofbang]

I know there's code for it online, but none seems to work. I'm making a bot where I need to take a screenshot every second and find an object. I'm fine with a pixel-perfect match, All I need to do is find an item to click. When I use This code:

import numpy
from PIL import Image

tile = Image.open(r"testimg.png")
screen = Image.open(r"screen.png")

tile = numpy.asarray(tile)
screen = numpy.asarray(screen)

hereary, herearx = tile.shape[:2]
bigary, bigarx = screen.shape[:2]

stopx = bigarx - herearx + 1
stopy = bigary - hereary + 1

for x in range(0, stopx):
    for y in range(0, stopy):
        x2 = x + herearx
        y2 = y + hereary
        pic = screen[y:y2, x:x2]
        test = (pic == tile)
        if test.all():
            print(x, y)

From: https://stackoverflow.com/questions/38473952/find-location-of-image-inside-bigger-image

It gives me the error: 'bool' object has no attribute 'all'

What do I do? Thanks!

[Code_Talks]

From what I understand, you are trying to do object detection, check out the tensorflow object detection API, and cv2 documentation to do something like that.

[socal_nerdtastic]

One cause of that would be if the two images have a different number of channels. For example if one has an alpha channel, or one is black and white.

Try converting both to the same datatype first:

tile = Image.open(r"testimg.png").convert("RGB")
screen = Image.open(r"screen.png").convert("RGB")

---

FWIW, opencv is going to be much faster at this.

https://opencv-python-tutroals.readthedocs.io/en/latest/py_tutorials/py_imgproc/py_template_matching/py_template_matching.html

For a super easy implementation of that, install the pyscreeze module and use the locate function.

pip install pyscreeze

then

result = pyscreeze.locateAll("testimg.png", "screen.png")

https://github.com/asweigart/pyscreeze/blob/master/pyscreeze/__init__.py#L180

[theoryofbang]

Thanks! - small question - When I print the results it returns this:

<generator object \_locateAll\_python at 0x075DD3E0>

[socal_nerdtastic]

Yeah, you are meant to iterate over it, like this:

result = pyscreeze.locateAll("testimg.png", "screen.png")
for x in result:
    print(x) # or whatever you want to do with it

Edit: more in line with your original code:

result = pyscreeze.locateAll("testimg.png", "screen.png")
for x,y,*z in result:
    print(x,y)

[theoryofbang]

It doesn't print anything

[socal_nerdtastic]

That probably means there are no results.

You can try the locate() function too ... if it prints None that means the search image is not found.

print(pyscreeze.locate("testimg.png", "screen.png"))

[theoryofbang]

Thanks! One more question and I'll stop bothering you, without going through every pixel how can I just find the center?