File size: 652 Bytes
3f613fd |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 |
# YOLOv5 🚀 by Ultralytics, GPL-3.0 license
"""
Perform test request
"""
import pprint
import cv2
import numpy as np
import matplotlib.pyplot as plt
import requests
DETECTION_URL = 'http://localhost:5000/v1/object-detection/yolov5s'
IMAGE = 'data/images/zidane.jpg'
# Read image
# with open(IMAGE, 'rb') as f:
# image_data = f.read()
img = cv2.imread(IMAGE)
img = cv2.cvtColor(img, cv2.COLOR_BGR2RGB)
img = cv2.imencode(".jpg", img)[1].tobytes()
response = requests.post(DETECTION_URL, data=img)
img = cv2.imdecode(np.frombuffer(response.content, dtype=np.uint8), cv2.IMREAD_COLOR)
plt.imshow(img)
plt.show()
# pprint.pprint(response)
|