Created
June 2, 2022 02:33
-
-
Save nakagami/12f3296f9246b26451dc168a89d9e684 to your computer and use it in GitHub Desktop.
get adult score by Microsoft Computer Vision API
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
import sys | |
import os | |
import json | |
import http.client | |
from urllib.parse import urlencode | |
# Computer Vision API | |
# https://azure.microsoft.com/en-us/services/cognitive-services/computer-vision/ | |
MICROSOFT_COMPUTER_VISION_API_KEY = os.environ['MICROSOFT_COMPUTER_VISION_API_KEY'] | |
def get_ms_vision_score(jpeg_path): | |
headers = { | |
'Content-Type': 'application/octet-stream', | |
'Ocp-Apim-Subscription-Key': MICROSOFT_COMPUTER_VISION_API_KEY, | |
} | |
params = urlencode({ | |
'visualFeatures': 'Adult', | |
}) | |
conn = http.client.HTTPSConnection('southeastasia.api.cognitive.microsoft.com') | |
conn.request("POST", "/vision/v1.0/analyze?%s" % params, open(jpeg_path, "br").read(), headers) | |
return conn.getresponse() | |
if __name__ == "__main__": | |
if len(sys.argv) < 2: | |
print("{} <image_file>".format(sys.argv[0])) | |
sys.exit() | |
response = get_ms_vision_score(sys.argv[1]) | |
print(json.dumps(json.loads(response.read().decode('utf-8')), indent=4)) |
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment