Local image tagging with Clarifai's Python client

clarifai_local_image_tagging.py

"""
Tagging a few local images using the Clarifai API and the clarifai-python
client.

"""

from os.path import expanduser
from clarifai.client import ClarifaiApi

directory = expanduser('~/pictures/clarifai/')
pictures = ['burning_basmatty.jpg', 'concert.jpg']

clarifai_api = ClarifaiApi()
results = clarifai_api.tag_images(
    [open('%s%s' % (directory, p), 'rb') for p in pictures]
)

# Uncomment the below to print the whole JSON returned
# from pprint import pprint
# pprint(result)

for i, result in enumerate(results['results']):
    tags = result['result']['tag']
    print '\nTags for %s (Tag - Probability)' % pictures[i]
    for j in range(len(tags['classes'])):
        print '%s - %0.4f' % (tags['classes'][j], tags['probs'][j])