deepgeek/MuseTalk
2
0
Fork 0
mirror of https://github.com/TMElyralab/MuseTalk.git synced 2026-02-04 09:29:20 +08:00
Code Issues Packages Projects Releases Wiki Activity

fix: preprocess import bug (#345)

Browse Source 
Fixing the issue when executing the script "python -m scripts.preprocess --config ./configs/training/preprocess.yaml" in the Ubuntu environment, due to the "import torch" order problem, the execution of "from musetalk.utils.face_detection import FaceAlignment, LandmarksType" hangs.
This commit is contained in:
Nick Davis
2025-07-02 16:40:49 +08:00
committed by GitHub
parent 26ca7c2c03
commit 6e39bd0d00
1 changed files with 4 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
scripts/preprocess.py
View File

@@ -1,6 +1,9 @@
import os import os
import argparse import argparse
import subprocess import subprocess
import torch
import numpy as np
from tqdm import tqdm
from omegaconf import OmegaConf from omegaconf import OmegaConf
from typing import Tuple, List, Union from typing import Tuple, List, Union
import decord import decord
@@ -9,9 +12,6 @@ import cv2
from musetalk.utils.face_detection import FaceAlignment,LandmarksType from musetalk.utils.face_detection import FaceAlignment,LandmarksType
from mmpose.apis import inference_topdown, init_model from mmpose.apis import inference_topdown, init_model
from mmpose.structures import merge_data_samples from mmpose.structures import merge_data_samples
import torch
import numpy as np
from tqdm import tqdm
import sys import sys
def fast_check_ffmpeg(): def fast_check_ffmpeg():
@@ -331,4 +331,4 @@ if __name__ == "__main__":
config = OmegaConf.load(args.config) config = OmegaConf.load(args.config)
main(config) main(config)