зеркало из https://github.com/Azure/azcv.git
85 строки
1.9 KiB
Python
85 строки
1.9 KiB
Python
# -*- coding: utf-8 -*-
|
|
|
|
# Copyright (c) Microsoft Corporation. All rights reserved.
|
|
# Licensed under the MIT License.
|
|
# Author: Graham.Williams@togaware.com
|
|
#
|
|
# A script to identify celebrities in a photo.
|
|
#
|
|
# ml celebrities azcv <path>
|
|
#
|
|
|
|
from azure.cognitiveservices.vision.computervision import ComputerVisionClient
|
|
from msrest.authentication import CognitiveServicesCredentials
|
|
from azure.cognitiveservices.vision.computervision.models import VisualFeatureTypes
|
|
|
|
import os
|
|
import sys
|
|
import argparse
|
|
|
|
from mlhub.pkg import azkey, is_url
|
|
from mlhub.utils import get_cmd_cwd
|
|
|
|
# ----------------------------------------------------------------------
|
|
# Parse command line arguments
|
|
# ----------------------------------------------------------------------
|
|
|
|
option_parser = argparse.ArgumentParser(add_help=False)
|
|
|
|
option_parser.add_argument(
|
|
'path',
|
|
help='path or url to image')
|
|
|
|
args = option_parser.parse_args()
|
|
|
|
# ----------------------------------------------------------------------
|
|
|
|
SERVICE = "Computer Vision"
|
|
KEY_FILE = os.path.join(os.getcwd(), "private.txt")
|
|
|
|
# Request subscription key and endpoint from user.
|
|
|
|
subscription_key, endpoint = azkey(KEY_FILE, SERVICE, verbose=False)
|
|
|
|
# Set credentials.
|
|
|
|
credentials = CognitiveServicesCredentials(subscription_key)
|
|
|
|
# Create client.
|
|
|
|
client = ComputerVisionClient(endpoint, credentials)
|
|
|
|
# Send image to azure to analyse.
|
|
|
|
url = args.path
|
|
|
|
features = VisualFeatureTypes.image_type
|
|
|
|
if is_url(url):
|
|
analysis = client.analyze_image(url, features)
|
|
else:
|
|
path = os.path.join(get_cmd_cwd(), url)
|
|
with open(path, 'rb') as fstream:
|
|
analysis = client.analyze_image_in_stream(fstream, features)
|
|
|
|
ca = analysis.image_type.clip_art_type
|
|
ld = analysis.image_type.line_drawing_type
|
|
|
|
cat = ""
|
|
if ca == 0:
|
|
cat = "no"
|
|
elif ca == 1:
|
|
cat = "ambiguous"
|
|
elif ca == 2:
|
|
cat = "ok"
|
|
elif ca == 3:
|
|
cat = "good"
|
|
|
|
ldt = ""
|
|
if ld == 0:
|
|
ldt = "no"
|
|
elif ld == 1:
|
|
ldt = "yes"
|
|
|
|
print(f"{cat},{ldt}")
|