--- a +++ b/demo-fall-detection.py @@ -0,0 +1,16 @@ +from PIL import Image +from fall_prediction import Fall_prediction + +img1 = Image.open("Images/fall_img_1.png") +img2 = Image.open("Images/fall_img_2.png") +img3 = Image.open("Images/fall_img_3.png") + +response = Fall_prediction(img1, img2, img3) + +if response: + print("There is", response['category']) + print("Confidence :", response['confidence']) + print("Angle : ", response['angle']) + print("Keypoint_corr :", response['keypoint_corr']) +else: + print("There is no fall detetcion...")