In [1]:
from selenium import webdriver
from IPython.display import Image, display
import time
In [29]:
# remove str
def removeStr(text, word):
return "".join(text.split(word))
def getVision(image_url, category="Labels"):
url = 'https://cloud.google.com/vision/'
# open site
driver = webdriver.Chrome()
driver.get(url)
# enter iframe
iframe = driver.find_element_by_css_selector("#vision_demo_section iframe")
driver.switch_to_frame(iframe)
# upload image
driver.find_element_by_id("input").send_keys(image_url)
# check loading done
delay = 0
for _ in range(30):
if driver.find_element_by_id("results").text != '':
break
time.sleep(1)
delay += 1
print("Analytics Delay Time : {} sec".format(delay))
# conditional category
if category == "Labels":
driver.find_element_by_css_selector("[data-type=labelAnnotations]").click()
time.sleep(3)
result = driver.find_element_by_id("results").text.split("\n")
elif category == "Web":
driver.find_element_by_css_selector("[data-type=webDetection]").click()
time.sleep(3)
result = driver.find_element_by_id("results").text.split("\n")[1:]
elif category == "Text":
driver.find_element_by_css_selector("[data-type=textAnnotations]").click()
time.sleep(3)
result = driver.find_element_by_css_selector("#results .text").text
result = removeStr(result, "“")
result = removeStr(result, "”")
result = result.strip()
# close driver
driver.close()
return result
In [30]:
def showResult(image_url, category):
result = getVision(image_url, category)
if category == "Text":
print(result)
else:
print(result[0], result[1], result[2], result[3], result[4])
In [4]:
image_url = '/Users/rada/Desktop/images/newyork.png'
category = 'Labels'
display(Image(filename = image_url))
showResult(image_url, category)
In [5]:
image_url = "/Users/rada/Desktop/images/jj.jpg"
category = "Web"
display(Image(filename = image_url))
showResult(image_url, category)
In [6]:
image_url = '/Users/rada/Desktop/images/ss.png'
category = "Web"
display(Image(filename = image_url))
showResult(image_url, category)
In [8]:
image_url = '/Users/rada/Desktop/images/ll.jpg'
category = "Web"
display(Image(filename = image_url))
showResult(image_url, category)
In [31]:
image_url = '/Users/rada/Desktop/images/text1.png'
category = "Text"
display(Image(filename = image_url))
showResult(image_url, category)
In [32]:
image_url = '/Users/rada/Desktop/images/text2.png'
category = "Text"
display(Image(filename = image_url))
showResult(image_url, category)
In [33]:
image_url = '/Users/rada/Desktop/images/text3.png'
category = "Text"
display(Image(filename = image_url))
showResult(image_url, category)
In [ ]: