Show FPS on demo video

rtmo_demo.py

@@ -38,7 +38,8 @@ if __name__ == "__main__":
             s = time.time()
             keypoints, scores = body(frame)
             det_time = time.time() - s
-            print(f'det: {round(1.0 / det_time,1)} FPS')
+            fps = round(1.0 / det_time,1)
+            print(f'det: {fps} FPS')
             
             img_show = frame.copy()
             
@@ -51,5 +52,6 @@ if __name__ == "__main__":
                                     kpt_thr=0.3,
                                     line_width=2)
             img_show = resize_to_fit_screen(img_show, 720, 480)
+            cv2.putText(img_show, f'{fps:.1f}', (10, 30), cv2.FONT_HERSHEY_COMPLEX_SMALL, 1, (0, 255, 0), 1, cv2.LINE_AA)
             cv2.imshow(f'{model}', img_show)
             cv2.waitKey(10)

rtmo_demo_batch.py

@@ -26,7 +26,8 @@ def process_video(video_path, body_estimator, batch_size=4):
             s = time.time()
             batch_keypoints, batch_scores = body_estimator(batch_frames)
             det_time = time.time() - s
-            print(f'Batch det: {round(batch_size / det_time, 1)} FPS')
+            fps = round(batch_size / det_time, 1)
+            print(f'Batch det: {fps} FPS')
 
             for i, keypoints in enumerate(batch_keypoints):
                 scores = batch_scores[i]
@@ -34,6 +35,7 @@ def process_video(video_path, body_estimator, batch_size=4):
                 img_show = frame.copy()
                 img_show = draw_skeleton(img_show, keypoints, scores, kpt_thr=0.3, line_width=2)
                 img_show = resize_to_fit_screen(img_show, 720, 480)
+                cv2.putText(img_show, f'{fps:.1f}', (10, 30), cv2.FONT_HERSHEY_COMPLEX_SMALL, 1, (0, 255, 0), 1, cv2.LINE_AA)
                 cv2.imshow(f'{video_path}', img_show)
                 cv2.waitKey(10)