Exemplo n.º 1
0
def test_logos(capsys):
    file_name = os.path.join(
        os.path.dirname(__file__),
        'resources/logos.png')
    detect.detect_logos(file_name)
    out, _ = capsys.readouterr()
    assert 'Google' in out
Exemplo n.º 2
0
from detect import (detect_logos, detect_text)
import pandas as pd
import re
import os
#from __future__ import print_function
from google.cloud import vision


images_path = "C:\\Users\\heinz\\Yagora GmbH\\Ievgen Kyrda - Crawler\\images\\foodnewsgermany_images/"
file_names = os.listdir(os.path.dirname(images_path))

file_paths = [images_path + f for f in file_names]

logos = [detect_logos(f) for f in file_paths]

texts = [detect_text(f)[0].description for f in file_paths]
# remove line break symbols
texts = [x.replace("\n", ", ") for x in texts]

contained = []
#contained[1] = "test"
for i in range(len(logos)): # loop over future rows of df
    tmp = []
    for j in logos[i]: # for every logo-row, check if in text
        if j.lower() in texts[i].lower():
            tmp.append(logos[i])
        else:
            tmp.append(None)
    contained.append(tmp)