Spaces:
Runtime error
Runtime error
Update roop/processors/frame/face_swapper.py
Browse files
roop/processors/frame/face_swapper.py
CHANGED
|
@@ -80,7 +80,7 @@ def process_frames(source_path: str, temp_frame_paths: List[str], update: Callab
|
|
| 80 |
def process_image(source_path: str, target_path: str, output_path: str) -> None:
|
| 81 |
source_face = get_one_face(cv2.imread(source_path))
|
| 82 |
target_frame = cv2.imread(target_path)
|
| 83 |
-
reference_face = None if roop.globals.many_faces else get_one_face(target_frame
|
| 84 |
result = process_frame(source_face, reference_face, target_frame)
|
| 85 |
cv2.imwrite(output_path, result)
|
| 86 |
|
|
@@ -88,6 +88,6 @@ def process_image(source_path: str, target_path: str, output_path: str) -> None:
|
|
| 88 |
def process_video(source_path: str, temp_frame_paths: List[str]) -> None:
|
| 89 |
if not roop.globals.many_faces and not get_face_reference():
|
| 90 |
reference_frame = cv2.imread(temp_frame_paths[roop.globals.reference_frame_number])
|
| 91 |
-
reference_face = get_one_face(reference_frame
|
| 92 |
set_face_reference(reference_face)
|
| 93 |
-
roop.processors.frame.core.process_video(source_path, temp_frame_paths, process_frames)
|
|
|
|
| 80 |
def process_image(source_path: str, target_path: str, output_path: str) -> None:
|
| 81 |
source_face = get_one_face(cv2.imread(source_path))
|
| 82 |
target_frame = cv2.imread(target_path)
|
| 83 |
+
reference_face = None if roop.globals.many_faces else get_one_face(target_frame)
|
| 84 |
result = process_frame(source_face, reference_face, target_frame)
|
| 85 |
cv2.imwrite(output_path, result)
|
| 86 |
|
|
|
|
| 88 |
def process_video(source_path: str, temp_frame_paths: List[str]) -> None:
|
| 89 |
if not roop.globals.many_faces and not get_face_reference():
|
| 90 |
reference_frame = cv2.imread(temp_frame_paths[roop.globals.reference_frame_number])
|
| 91 |
+
reference_face = get_one_face(reference_frame)
|
| 92 |
set_face_reference(reference_face)
|
| 93 |
+
roop.processors.frame.core.process_video(source_path, temp_frame_paths, process_frames)
|