fix paths

app.py

@@ -138,7 +138,7 @@ class App:
 if __name__ == "__main__":
     parser = argparse.ArgumentParser()
     parser.add_argument('--model_dir', type=str, default=os.path.join("pretrained_weights"), help='Pretrained models directory for MusePose')
-    parser.add_argument('--output_dir', type=str, default=os.path.join("assets", "videos"), help='Output directory for the result')
+    parser.add_argument('--output_dir', type=str, default=os.path.join("outputs"), help='Output directory for the result')
     parser.add_argument('--disable_model_download_at_start', type=bool, default=False, nargs='?', const=True, help='Disable model download at start or not')
     args = parser.parse_args()
 

musepose_inference.py

@@ -83,8 +83,8 @@ class MusePoseInference:
         image_file_name = os.path.splitext(os.path.basename(ref_image_path))[0]
         pose_video_file_name = os.path.splitext(os.path.basename(pose_video_path))[0]
         output_file_name = f"img_{image_file_name}_pose_{pose_video_file_name}"
-        output_path = os.path.abspath(os.path.join(self.output_dir, f'{output_file_name}.mp4'))
-        output_path_demo = os.path.abspath(os.path.join(self.output_dir, f'{output_file_name}_demo.mp4'))
+        output_path = os.path.abspath(os.path.join(self.output_dir, "musepose_inference", f'{output_file_name}.mp4'))
+        output_path_demo = os.path.abspath(os.path.join(self.output_dir, "musepose_inference", f'{output_file_name}_demo.mp4'))
 
         if weight_dtype == "fp16":
             weight_dtype = torch.float16

pose_align.py

@@ -46,9 +46,10 @@ class PoseAlignmentInference:
         max_frame: int,
     ):
         download_models(model_dir=self.model_dir)
-        dt_file_name = datetime.now().strftime("%Y-%m-%d_%H-%M-%S")
-        outfn=os.path.abspath(os.path.join(self.output_dir, f'{dt_file_name}_demo.mp4'))
-        outfn_align_pose_video=os.path.abspath(os.path.join(self.output_dir, f'{dt_file_name}.mp4'))
+        img_file_name = os.path.splitext(os.path.basename(imgfn_refer))[0]
+        vid_file_name = os.path.splitext(os.path.basename(vidfn))[0]
+        outfn=os.path.abspath(os.path.join(self.output_dir, "pose_alignment", f'img_{img_file_name}_vid_{vid_file_name}_demo.mp4'))
+        outfn_align_pose_video=os.path.abspath(os.path.join(self.output_dir, "pose_alignment", f'img_{img_file_name}_vid_{vid_file_name}.mp4'))
 
         video = cv2.VideoCapture(vidfn)
         width= video.get(cv2.CAP_PROP_FRAME_WIDTH)