| # 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) | |