Update demo.py

Fixed bugs
Sudiksha18 · May 12, 2024 · ce3662d · ce3662d
1 parent 6a3faef
commit ce3662d
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/Suicide_Detection/demo.py b/Suicide_Detection/demo.py
@@ -51,10 +51,10 @@
             for box in b:
                 if box.any():
                     _x, _y, _w, _h = box
-                    x1 = round(_x)
-                    y1 = round(_y)
-                    wo = round(_w)
-                    ho = round(_h)
+                    x1 = round(_x * w)
+                    y1 = round(_y * h)
+                    wo = round(_w * w)
+                    ho = round(_h * h)
 
                 cv2.rectangle(frame, (x1, y1), (x1 + wo, y1 + ho), (255, 0, 0), 2)