edit import

Demo_TFR_Pirenderer/src/audio2pose_models/audio2pose.py

@@ -1,8 +1,8 @@
 import torch
 from torch import nn
-from src.audio2pose_models.cvae import CVAE
-from src.audio2pose_models.discriminator import PoseSequenceDiscriminator
-from src.audio2pose_models.audio_encoder import AudioEncoder
+from Demo_TFR_Pirenderer.src.audio2pose_models.cvae import CVAE
+from Demo_TFR_Pirenderer.src.audio2pose_models.discriminator import PoseSequenceDiscriminator
+from Demo_TFR_Pirenderer.src.audio2pose_models.audio_encoder import AudioEncoder
 
 class Audio2Pose(nn.Module):
     def __init__(self, cfg, wav2lip_checkpoint, device='cuda'):

Demo_TFR_Pirenderer/src/audio2pose_models/cvae.py

@@ -1,7 +1,7 @@
 import torch
 import torch.nn.functional as F
 from torch import nn
-from src.audio2pose_models.res_unet import ResUnet
+from Demo_TFR_Pirenderer.src.audio2pose_models.res_unet import ResUnet
 
 def class2onehot(idx, class_num):
 

Demo_TFR_Pirenderer/src/audio2pose_models/res_unet.py

@@ -1,6 +1,6 @@
 import torch
 import torch.nn as nn
-from src.audio2pose_models.networks import ResidualConv, Upsample
+from Demo_TFR_Pirenderer.src.audio2pose_models.networks import ResidualConv, Upsample
 
 
 class ResUnet(nn.Module):

Demo_TFR_Pirenderer/src/face3d/models/__init__.py

@@ -19,7 +19,7 @@ See our template model class 'template_model.py' for more details.
 """
 
 import importlib
-from src.face3d.models.base_model import BaseModel
+from Demo_TFR_Pirenderer.src.face3d.models.base_model import BaseModel
 
 
 def find_model_using_name(model_name):

Demo_TFR_Pirenderer/src/face3d/models/bfm.py

@@ -5,7 +5,7 @@ import numpy as np
 import  torch
 import torch.nn.functional as F
 from scipy.io import loadmat
-from src.face3d.util.load_mats import transferBFM09
+from Demo_TFR_Pirenderer.src.face3d.util.load_mats import transferBFM09
 import os
 
 def perspective_projection(focal, center):

Demo_TFR_Pirenderer/src/face3d/models/facerecon_model.py

@@ -3,13 +3,13 @@
 
 import numpy as np
 import torch
-from src.face3d.models.base_model import BaseModel
-from src.face3d.models import networks
-from src.face3d.models.bfm import ParametricFaceModel
-from src.face3d.models.losses import perceptual_loss, photo_loss, reg_loss, reflectance_loss, landmark_loss
-from src.face3d.util import util 
-from src.face3d.util.nvdiffrast import MeshRenderer
-# from src.face3d.util.preprocess import estimate_norm_torch
+from Demo_TFR_Pirenderer.src.face3d.models.base_model import BaseModel
+from Demo_TFR_Pirenderer.src.face3d.models import networks
+from Demo_TFR_Pirenderer.src.face3d.models.bfm import ParametricFaceModel
+from Demo_TFR_Pirenderer.src.face3d.models.losses import perceptual_loss, photo_loss, reg_loss, reflectance_loss, landmark_loss
+from Demo_TFR_Pirenderer.src.face3d.util import util
+from Demo_TFR_Pirenderer.src.face3d.util.nvdiffrast import MeshRenderer
+# from Demo_TFR_Pirenderer.src.face3d.util.preprocess import estimate_norm_torch
 
 import trimesh
 from scipy.io import savemat

Demo_TFR_Pirenderer/src/face3d/util/__init__.py

@@ -1,3 +1,3 @@
 """This package includes a miscellaneous collection of useful helper functions."""
-from src.face3d.util import *
+from Demo_TFR_Pirenderer.src.face3d.util import *
 

Demo_TFR_Pirenderer/src/face3d/visualize.py

@@ -1,8 +1,8 @@
 # check the sync of 3dmm feature and the audio
 import cv2
 import numpy as np
-from src.face3d.models.bfm import ParametricFaceModel
-from src.face3d.models.facerecon_model import FaceReconModel
+from Demo_TFR_Pirenderer.src.face3d.models.bfm import ParametricFaceModel
+from Demo_TFR_Pirenderer.src.face3d.models.facerecon_model import FaceReconModel
 import torch
 import subprocess, platform
 import scipy.io as scio

Demo_TFR_Pirenderer/src/pirenderer/animate.py

@@ -13,8 +13,8 @@ import torch
 import torchvision
 
 from pydub import AudioSegment
-from src.pirenderer.modules.face_model import FaceGenerator
-from src.utils.videoio import save_video_with_watermark
+from Demo_TFR_Pirenderer.src.pirenderer.modules.face_model import FaceGenerator
+from Demo_TFR_Pirenderer.src.utils.videoio import save_video_with_watermark
 
 class AnimateFromCoeff():
 

Demo_TFR_Pirenderer/src/pirenderer/modules/face_model.py

@@ -7,8 +7,8 @@ import torch.nn.functional as F
 import sys
 sys.path.append("..")
 
-from src.pirenderer.util import flow_util
-from src.pirenderer.modules.base_function import  LayerNorm2d, ADAINHourglass, FineEncoder, FineDecoder
+from Demo_TFR_Pirenderer.src.pirenderer.util import flow_util
+from Demo_TFR_Pirenderer.src.pirenderer.modules.base_function import  LayerNorm2d, ADAINHourglass, FineEncoder, FineDecoder
 
 
 class FaceGenerator(nn.Module):

Demo_TFR_Pirenderer/src/test_audio2coeff.py

@@ -5,9 +5,9 @@ from scipy.io import savemat, loadmat
 from yacs.config import CfgNode as CN
 from scipy.signal import savgol_filter
 
-from src.audio2pose_models.audio2pose import Audio2Pose
-from src.audio2exp_models.networks import SimpleWrapperV2 
-from src.audio2exp_models.audio2exp import Audio2Exp  
+from Demo_TFR_Pirenderer.src.audio2pose_models.audio2pose import Audio2Pose
+from Demo_TFR_Pirenderer.src.audio2exp_models.networks import SimpleWrapperV2
+from Demo_TFR_Pirenderer.src.audio2exp_models.audio2exp import Audio2Exp
 
 def load_cpk(checkpoint_path, model=None, optimizer=None, device="cpu"):
     checkpoint = torch.load(checkpoint_path, map_location=torch.device(device))

Demo_TFR_Pirenderer/src/utils/audio.py

@@ -4,7 +4,7 @@ import numpy as np
 # import tensorflow as tf
 from scipy import signal
 from scipy.io import wavfile
-from src.utils.hparams import hparams as hp
+from Demo_TFR_Pirenderer.src.utils.hparams import hparams as hp
 
 def load_wav(path, sr):
     return librosa.core.load(path, sr=sr)[0]

Demo_TFR_Pirenderer/src/utils/face_enhancer.py

@@ -5,7 +5,7 @@ from gfpgan import GFPGANer
 
 from tqdm import tqdm
 
-from src.utils.videoio import load_video_to_cv2
+from Demo_TFR_Pirenderer.src.utils.videoio import load_video_to_cv2
 
 import cv2
 

Demo_TFR_Pirenderer/src/utils/paste_pic.py

@@ -3,7 +3,7 @@ import numpy as np
 from tqdm import tqdm
 import uuid
 
-from src.utils.videoio import save_video_with_watermark 
+from Demo_TFR_Pirenderer.src.utils.videoio import save_video_with_watermark
 
 def paste_pic(video_path, pic_path, crop_info, new_audio_path, full_video_path):
 

Demo_TFR_Pirenderer/src/utils/preprocess.py

@@ -4,19 +4,19 @@ from tqdm import tqdm
 from PIL import Image 
 
 # 3dmm extraction
-from src.face3d.util.preprocess import align_img
-from src.face3d.util.load_mats import load_lm3d
-from src.face3d.models import networks
+from Demo_TFR_Pirenderer.src.face3d.util.preprocess import align_img
+from Demo_TFR_Pirenderer.src.face3d.util.load_mats import load_lm3d
+from Demo_TFR_Pirenderer.src.face3d.models import networks
 
 try:
     import webui
-    from src.face3d.extract_kp_videos_safe import KeypointExtractor
+    from Demo_TFR_Pirenderer.src.face3d.extract_kp_videos_safe import KeypointExtractor
     assert torch.cuda.is_available() == True
 except:
-    from src.face3d.extract_kp_videos import KeypointExtractor
+    from Demo_TFR_Pirenderer.src.face3d.extract_kp_videos import KeypointExtractor
 
 from scipy.io import loadmat, savemat
-from src.utils.croper import Croper
+from Demo_TFR_Pirenderer.src.utils.croper import Croper
 
 import warnings 
 warnings.filterwarnings("ignore")