WKO_PROJEKT/yolov5/utils/flask_rest_api/example_request.py

20 lines
368 B
Python
Raw Permalink Normal View History

2023-02-01 23:02:19 +01:00
# YOLOv5 🚀 by Ultralytics, GPL-3.0 license
"""
Perform test request
"""
import pprint
import requests
DETECTION_URL = "http://localhost:5000/v1/object-detection/yolov5s"
IMAGE = "zidane.jpg"
# Read image
with open(IMAGE, "rb") as f:
image_data = f.read()
response = requests.post(DETECTION_URL, files={"image": image_data}).json()
pprint.pprint(response)