From e52ba46512aa0c4b37df161cbccca7fa11842088 Mon Sep 17 00:00:00 2001
From: zaqxs123456 <56287342+zaqxs123456@users.noreply.github.com>
Date: Fri, 25 Oct 2024 10:03:23 +0800
Subject: [PATCH] add deviation

---
 app.py | 11 ++++++++++-
 1 file changed, 10 insertions(+), 1 deletion(-)

diff --git a/app.py b/app.py
index be43a07..f9586bd 100644
--- a/app.py
+++ b/app.py
@@ -79,7 +79,16 @@ def get_predicted_coordinates(coordinates: list, width: int, height: int) -> lis
     # those are in skeleton_lib.py
 
     # when testing, can visualize with the method expo_save_bodypose in openpose_gen.py
-    return mirror_coordinates(coordinates, width)
+
+    # for now, just mirror the coordinates and add some random deviation
+    predicted_coordinates = mirror_coordinates(coordinates, width)
+    for i in range(0, len(predicted_coordinates), 3):
+        deviation = random.randint(-10, 10)
+        predicted_coordinates[i] += deviation
+        deviation = random.randint(-10, 10)
+        predicted_coordinates[i + 1] += deviation
+
+    return predicted_coordinates
 
 def mirror_coordinates(coordinates: list, width: int) -> list:
     for i in range(0, len(coordinates), 3):