Fixed minor issues

RizwanMunawar · Nov 22, 2022 · 60e65b2 · 60e65b2
1 parent e462e4e
commit 60e65b2
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/pose-estimate.py b/pose-estimate.py
@@ -115,6 +115,9 @@ def run(poseweights="yolov7-w6-pose.pt",source="football1.mp4",device='cpu',view
 
                 out.write(im0)  #writing the video frame
 
+            else:
+                break
+
         cap.release()
         # cv2.destroyAllWindows()
         avg_fps = total_fps / frame_count