import argparse
from oauth2client.client import GoogleCredentials
from googleapiclient import discovery
from googleapiclient import errors
from PIL import Image

parser = argparse.ArgumentParser()
parser.add_argument('-k', '--keep_top_k', type=int, default=10)
parser.add_argument('-c', '--confidence_threshold', type=float, default=0.8)
parser.add_argument('-i', '--image', required=True)
args = parser.parse_args()

project = YOUR_PROJECT_ID
model   = 'ssd_mlengine'
version = YOUR_MODEL_VERSION

credentials = GoogleCredentials.get_application_default()
ml = discovery.build('ml', 'v1', credentials=credentials)

size = (300, 300)
img = Image.open(args.image)
original_size = img.size
resized_img = img.resize(size)

data = []
for i in range(size[1]):
    data.append([])
    for j in range(size[0]):
        data[i].append(resized_img.getpixel((j, i)))

body = {'instances': [{'key': '1',
    'data': data,
    'keep_top_k': args.keep_top_k,
    'original_size': original_size,
    'confidence_threshold': args.confidence_threshold
    }]}
request = ml.projects().predict(name='projects/{}/models/{}/versions/{}'.format(project, model, version), body=body)
try:
    response = request.execute()
    output = response['predictions']
    print(output)
except errors.HttpError as err:
    print(err._get_reason())