Add single_frame mode for webcam

2024-07-21 19:42:22 +03:00
parent 5e3c92d55c
commit 4cefac79b8
2 changed files with 6 additions and 0 deletions
@@ -174,6 +174,11 @@ class LivePortraitPipeline(object):
                delta_new = x_s_info['exp']
                scale_new = x_s_info["scale"]
                t_new = x_d_info["t"]
+            elif relative_motion_mode == "single_frame":
+                R_new = R_d
+                delta_new = x_d_info['exp']
+                scale_new = x_s_info["scale"]
+                t_new = x_d_info["t"]
            else:
                R_new = R_d
                delta_new = x_s_info['exp']
@@ -258,6 +258,7 @@ class LivePortraitProcess:
                        "relative",
                        "source_video_smoothed",
                        "relative_rotation_only",
+                        "single_frame",
                        "off"
                    ],
                ),