forked from watson-developer-cloud/python-sdk
-
Notifications
You must be signed in to change notification settings - Fork 1
Expand file tree
/
Copy pathvisual_recognition_v3.py
More file actions
63 lines (50 loc) · 2.44 KB
/
visual_recognition_v3.py
File metadata and controls
63 lines (50 loc) · 2.44 KB
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
import json
from os.path import join, dirname
from watson_developer_cloud import VisualRecognitionV3
test_url = 'https://www.ibm.com/ibm/ginni/images' \
'/ginni_bio_780x981_v4_03162016.jpg'
visual_recognition = VisualRecognitionV3('2016-05-20', api_key='YOUR API KEY')
# with open(join(dirname(__file__), '../resources/cars.zip'), 'rb') as cars, \
# open(join(dirname(__file__), '../resources/trucks.zip'), 'rb') as
# trucks:
# print(json.dumps(visual_recognition.create_classifier('Cars vs Trucks',
# cars_positive_examples=cars,
#
# negative_examples=trucks), indent=2))
car_path = join(dirname(__file__), '../resources/cars.zip')
with open(car_path, 'rb') as images_file:
car_results = visual_recognition.classify(images_file=images_file,
threshold=0.1,
classifier_ids=[
'CarsvsTrucks_1479118188',
'default'])
print(json.dumps(car_results, indent=2))
# print(json.dumps(visual_recognition.get_classifier('YOUR CLASSIFIER ID'),
# indent=2))
# with open(join(dirname(__file__), '../resources/car.jpg'), 'rb') as
# image_file:
# print(json.dumps(visual_recognition.update_classifier(
# 'CarsvsTrucks_1479118188',
#
# cars_positive_examples=image_file), indent=2))
url_result = visual_recognition.classify(images_url=test_url)
print(json.dumps(url_result, indent=2))
faces_result = visual_recognition.detect_faces(images_url=test_url)
print(json.dumps(faces_result, indent=2))
# print(json.dumps(visual_recognition.delete_classifier(classifier_id='YOUR
# CLASSIFIER ID'), indent=2))
print(json.dumps(visual_recognition.list_classifiers(), indent=2))
file_path = join(dirname(__file__), '../resources/text.png')
with open(file_path, 'rb') as image_file:
text_results = visual_recognition.recognize_text(images_file=image_file)
print(json.dumps(text_results, indent=2))
face_path = join(dirname(__file__), '../resources/face.jpg')
with open(face_path, 'rb') as image_file:
face_result = visual_recognition.detect_faces(images_file=image_file)
print(json.dumps(face_result, indent=2))
# with open(join(dirname(__file__), '../resources/face.jpg'), 'rb') as \
# image_file:
# print(json.dumps(
# visual_recognition.find_similar(collection_id="YOUR_COLLECTION_ID",
# image_file=image_file),
# indent=2))