Added max_fps feature to optionally limit fps

afy/cam_fomm.py

@@ -29,26 +29,26 @@
 def is_new_frame_better(source, driving, predictor):
     global avatar_kp
     global display_string
-    
+
     if avatar_kp is None:
         display_string = "No face detected in avatar."
         return False
-    
+
     if predictor.get_start_frame() is None:
         display_string = "No frame to compare to."
         return True
-    
+
     driving_smaller = resize(driving, (128, 128))[..., :3]
     new_kp = predictor.get_frame_kp(driving)
-    
+
     if new_kp is not None:
         new_norm = (np.abs(avatar_kp - new_kp) ** 2).sum()
         old_norm = (np.abs(avatar_kp - predictor.get_start_frame_kp()) ** 2).sum()
-        
+
         out_string = "{0} : {1}".format(int(new_norm * 100), int(old_norm * 100))
         display_string = out_string
         log(out_string)
-        
+
         return new_norm < old_norm
     else:
         display_string = "No face found!"
@@ -183,6 +183,15 @@ def select_camera(config):
 if __name__ == "__main__":
     with open('config.yaml', 'r') as f:
         config = yaml.load(f, Loader=yaml.FullLoader)
+        if not 'max_fps' in config:
+            f.seek(0)
+            data = f.read(10 ** 4).rstrip("\n")
+            config['max_fps'] = None
+        max_fps = config['max_fps']
+    if config['max_fps'] == None:
+        with open('config.yaml', 'w') as f:
+            f.write(f"{data}\n\n# limit cam fps by setting this value to a positive integer\nmax_fps: 0")
+        max_fps = config['max_fps'] = 0
 
     global display_string
     display_string = ""
@@ -245,11 +254,11 @@ def select_camera(config):
         else:
             enable_vcam = False
             # log("Virtual camera is supported only on Linux.")
-        
+
         # if not enable_vcam:
             # log("Virtual camera streaming will be disabled.")
 
-    cur_ava = 0    
+    cur_ava = 0
     avatar = None
     change_avatar(predictor, avatars[cur_ava])
     passthrough = False
@@ -278,6 +287,7 @@ def select_camera(config):
     try:
         while True:
             tt = TicToc()
+            tt.tic()
 
             timing = {
                 'preproc': 0,
@@ -287,7 +297,6 @@ def select_camera(config):
 
             green_overlay = False
 
-            tt.tic()
 
             ret, frame = cap.read()
             if not ret:
@@ -321,7 +330,7 @@ def select_camera(config):
                 out = None
 
             tt.tic()
-            
+
             key = cv2.waitKey(1)
 
             if cv2.getWindowProperty('cam', cv2.WND_PROP_VISIBLE) < 1.0:
@@ -375,7 +384,7 @@ def select_camera(config):
                 if not is_calibrated:
                     cv2.namedWindow('avatarify', cv2.WINDOW_GUI_NORMAL)
                     cv2.moveWindow('avatarify', 600, 250)
-                
+
                 is_calibrated = True
                 show_landmarks = False
             elif key == ord('z'):
@@ -429,18 +438,18 @@ def select_camera(config):
                 draw_face_landmarks(preview_frame, avatar_kp, (200, 20, 10))
                 frame_kp = predictor.get_frame_kp(frame)
                 draw_face_landmarks(preview_frame, frame_kp)
-            
+
             if preview_flip:
                 preview_frame = cv2.flip(preview_frame, 1)
-                
+
             if green_overlay:
                 green_alpha = 0.8
                 overlay = preview_frame.copy()
                 overlay[:] = (0, 255, 0)
                 preview_frame = cv2.addWeighted( preview_frame, green_alpha, overlay, 1.0 - green_alpha, 0.0)
 
             timing['postproc'] = tt.toc()
-                
+
             if find_keyframe:
                 preview_frame = cv2.putText(preview_frame, display_string, (10, 220), 0, 0.5 * IMG_SIZE / 256, (255, 255, 255), 1)
 
@@ -470,10 +479,17 @@ def select_camera(config):
 
                 cv2.imshow('avatarify', out[..., ::-1])
 
+            if max_fps >= 1:
+                lapse = tt.toc(total=True) / 1000
+                sleep = (1 / max_fps) - lapse
+                if sleep > 0:
+                    time.sleep(sleep)
+
             fps_hist.append(tt.toc(total=True))
             if len(fps_hist) == 10:
                 fps = 10 / (sum(fps_hist) / 1000)
                 fps_hist = []
+
     except KeyboardInterrupt:
         log("main: user interrupt")