From f5579a95bd207135fb6316c3462e9421ed7f17d4 Mon Sep 17 00:00:00 2001
From: itechmusic <163980830+itechmusic@users.noreply.github.com>
Date: Thu, 18 Apr 2024 18:03:58 +0800
Subject: [PATCH] Fix fps calculation bug in realtime_inference.py (#35)

---
 scripts/realtime_inference.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/scripts/realtime_inference.py b/scripts/realtime_inference.py
index 07874c9..e5b7765 100644
--- a/scripts/realtime_inference.py
+++ b/scripts/realtime_inference.py
@@ -205,7 +205,7 @@ class Avatar:
             #combine_frame = get_image(ori_frame,res_frame,bbox)
             combine_frame = get_image_blending(ori_frame,res_frame,bbox,mask,mask_crop_box)
 
-            fps = 1/(time.time()-start)
+            fps = 1/(time.time()-start+1e-6)
             print(f"Generating the {self.idx}-th frame with FPS: {fps:.2f}")
             cv2.imwrite(f"{self.avatar_path}/tmp/{str(self.idx).zfill(8)}.png",combine_frame)
             self.idx = self.idx + 1
@@ -292,4 +292,4 @@ if __name__ == "__main__":
         for audio_num, audio_path in audio_clips.items():
             print("Inferring using:",audio_path)
             avatar.inference(audio_path, audio_num, args.fps)
-        
\ No newline at end of file
+