Skip to content

Commit 8ad8c7f

Browse files
authored
test(VR) core ml example (#428)
1 parent 7a4397b commit 8ad8c7f

2 files changed

Lines changed: 8 additions & 2 deletions

File tree

examples/visual_recognition_v3.py

Lines changed: 7 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -7,6 +7,7 @@
77
'/ginni_bio_780x981_v4_03162016.jpg'
88

99
visual_recognition = VisualRecognitionV3('2016-05-20', api_key='YOUR API KEY')
10+
classifier_id = 'CarsvsTrucksxDO_NOT_DELETE_771019274'
1011

1112
# with open(join(dirname(__file__), '../resources/cars.zip'), 'rb') as cars, \
1213
# open(join(dirname(__file__), '../resources/trucks.zip'), 'rb') as
@@ -18,7 +19,7 @@
1819

1920
car_path = join(dirname(__file__), '../resources/cars.zip')
2021
with open(car_path, 'rb') as images_file:
21-
parameters = json.dumps({'threshold': 0.1, 'classifier_ids': ['CarsvsTrucks_931077904', 'default']})
22+
parameters = json.dumps({'threshold': 0.1, 'classifier_ids': [classifier_id, 'default']})
2223
car_results = visual_recognition.classify(images_file=images_file,
2324
parameters=parameters)
2425
print(json.dumps(car_results, indent=2))
@@ -53,3 +54,8 @@
5354
with open(face_path, 'rb') as image_file:
5455
face_result = visual_recognition.detect_faces(images_file=image_file)
5556
print(json.dumps(face_result, indent=2))
57+
58+
#Core ml model example
59+
core_ml_model = visual_recognition.get_core_ml_model(classifier_id)
60+
with open('/tmp/CarsvsTrucks.mlmodel', 'wb') as fp:
61+
fp.write(core_ml_model.content)

test/integration/test_visual_recognition.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -19,7 +19,7 @@ def setUp(self):
1919
'X-Watson-Test':
2020
'1'
2121
})
22-
self.classifier_id = 'CarsvsTrucks_931077904'
22+
self.classifier_id = 'CarsvsTrucksxDO_NOT_DELETE_771019274'
2323

2424
def test_classify(self):
2525
car_path = join(dirname(__file__), '../../resources/cars.zip')

0 commit comments

Comments
 (0)