Prompt friendly messages if the docker_torchserve returns an error #…

…112 #137 #141

examples/image_to_annotations.py

@@ -49,6 +49,9 @@ def image_to_annotations(img_fn: str, out_dir: str) -> None:
     img_b = cv2.imencode('.png', img)[1].tobytes()
     request_data = {'data': img_b}
     resp = requests.post("http://localhost:8080/predictions/drawn_humanoid_detector", files=request_data, verify=False)
+    if resp is None or resp.status_code >= 300:
+        raise Exception(f"Failed to get bounding box, please check if the 'docker_torchserve' is running and healthy, resp: {resp}")
+
     detection_results = json.loads(resp.content)
 
     # error check detection_results
@@ -90,6 +93,9 @@ def image_to_annotations(img_fn: str, out_dir: str) -> None:
     # send cropped image to pose estimator
     data_file = {'data': cv2.imencode('.png', cropped)[1].tobytes()}
     resp = requests.post("http://localhost:8080/predictions/drawn_humanoid_pose_estimator", files=data_file, verify=False)
+    if resp is None or resp.status_code >= 300:
+        raise Exception(f"Failed to get skeletons, please check if the 'docker_torchserve' is running and healthy, resp: {resp}")
+
     pose_results = json.loads(resp.content)
 
     # error check pose_results