add: count

app.py

@@ -5,11 +5,16 @@ import os
 # ffmpeg -i input_videos/kun_1280x720_30fps_0-14_0-32.mp4 -vf trim=0:5,setpts=PTS-STARTPTS input_videos/kun_test_5sec.mp4
 # ffmpeg -i input.mp4 -vf scale=320:-1 output.mp4
 
+Count = 0
 
 def Video2MC(video, progress=gr.Progress(track_tqdm=True)):
 
     progress(1.0, desc="Step 0: Starting")
     output_path, output_video = gr_video2mc(video, progress)
+    
+    global Count 
+    Count += 1
+    print(f"Count: {Count}")
 
     return output_path, output_path, output_video
 
@@ -125,6 +130,8 @@ with gr.Blocks() as iface:
 
                             使用前，强烈建议先观看我的[B站视频](https://www.bilibili.com/video/BV1SP411W7pw)，快速了解该项目的用法。
 
+                            注：目前使用的是CPU，代码运行较慢。GPU算力过于昂贵，若有需求请私信联系。
+
                             """
                             )
         

videopose_PSTMO.py

@@ -14,6 +14,10 @@ from model.stmo import Model
 
 import HPE2keyframes as Hk 
 
+from datetime import datetime
+import pytz
+
+
 # from joints_detectors.openpose.main import generate_kpts as open_pose
 
 
@@ -198,6 +202,12 @@ def inference_video(video_path, detector_2d, progress):
 
 def gr_video2mc(video_path, progress):
 
+    print("\n>>>>> One video uploaded <<<<<\n")
+    china_tz = pytz.timezone('Asia/Shanghai')
+    current_time = datetime.now(china_tz)
+    formatted_time = current_time.strftime('%Y-%m-%d %H:%M:%S')
+    print(f"Start Time: {formatted_time}\n")
+
     if not os.path.exists('output_3Dpose_npy'):
         os.makedirs('output_3Dpose_npy')
     if not os.path.exists('output_alphapose'):
@@ -212,6 +222,14 @@ def gr_video2mc(video_path, progress):
     Hk.hpe2keyframes(output_dir_dict['npy'], FPS_mine_imator, f"output_miframes/{output_dir_dict['video_name']}.miframes")
     path1 = os.path.abspath(f"output_miframes/{output_dir_dict['video_name']}.miframes")
     path2 = os.path.abspath(f"output_videos/{output_dir_dict['video_name']}.mp4")
+
+    print("\n----- One video processed -----\n")
+    china_tz = pytz.timezone('Asia/Shanghai')
+    current_time = datetime.now(china_tz)
+    formatted_time = current_time.strftime('%Y-%m-%d %H:%M:%S')
+    print(f"Finished Time: {formatted_time}\n")
+
+
     return path1, path2