From 21991d51216a9e080db9f685553395f8e8799506 Mon Sep 17 00:00:00 2001 From: Anirban Mukherjee Date: Tue, 19 Dec 2023 14:12:20 +0530 Subject: [PATCH 1/2] Update inference.py - Float types - yaml load --- inference.py | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/inference.py b/inference.py index ed479e8..d98c36b 100644 --- a/inference.py +++ b/inference.py @@ -40,7 +40,7 @@ def draw_annotation_box( image, rotation_vector, translation_vector, color=(255, point_3d.append((front_size, front_size, front_depth)) point_3d.append((front_size, -front_size, front_depth)) point_3d.append((-front_size, -front_size, front_depth)) - point_3d = np.array(point_3d, dtype=np.float).reshape(-1, 3) + point_3d = np.array(point_3d, dtype=float).reshape(-1, 3) # Map to 2d image points (point_2d, _) = cv2.projectPoints(point_3d, @@ -138,7 +138,7 @@ def audio2head(audio_path, img_path, model_path, save_path): config_file = r"./config/vox-256.yaml" with open(config_file) as f: - config = yaml.load(f) + config = yaml.safe_load(f) kp_detector = KPDetector(**config['model_params']['kp_detector_params'], **config['model_params']['common_params']) generator = OcclusionAwareGenerator(**config['model_params']['generator_params'], @@ -146,7 +146,7 @@ def audio2head(audio_path, img_path, model_path, save_path): kp_detector = kp_detector.cuda() generator = generator.cuda() - opt = argparse.Namespace(**yaml.load(open("./config/parameters.yaml"))) + opt = argparse.Namespace(**yaml.safe_load(open("./config/parameters.yaml"))) audio2kp = AudioModel3D(opt).cuda() checkpoint = torch.load(model_path) @@ -248,4 +248,4 @@ def audio2head(audio_path, img_path, model_path, save_path): parse = parser.parse_args() os.makedirs(parse.save_path,exist_ok=True) - audio2head(parse.audio_path,parse.img_path,parse.model_path,parse.save_path) \ No newline at end of file + audio2head(parse.audio_path,parse.img_path,parse.model_path,parse.save_path) From c7e3dece7264119bf27439f8d57e48cbfadcae45 Mon Sep 17 00:00:00 2001 From: Anirban Mukherjee Date: Thu, 15 Feb 2024 12:49:36 +0530 Subject: [PATCH 2/2] Update inference.py --- inference.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/inference.py b/inference.py index d98c36b..2771a18 100644 --- a/inference.py +++ b/inference.py @@ -1,3 +1,5 @@ +### INFERENCE.PY: + import argparse import subprocess import python_speech_features