speed

app.py

@@ -9,10 +9,10 @@ css = """
 
 from huggingface_hub import snapshot_download
 from motion.visual_api import Visualize  
-import cv2
 import torch
 import json
 from tqdm import tqdm
+import imageio
 
 with open("motion/path.json", "r") as f:
     json_dict = json.load(f)
@@ -73,12 +73,7 @@ def demo(prompt, mode, condition, render_mode="joints", skip_steps=0, out_size=1
             meshes, _ = visual.get_mesh(output)
             pics = visual.pyrender_process(meshes, out_size, out_size)
         
-        video=cv2.VideoWriter(video_path,cv2.VideoWriter_fourcc(*'MP4V'),20,(out_size, out_size))
-        for pic in tqdm(pics):
-            pic = pic[:, :, :3]
-            pic = cv2.cvtColor(pic, cv2.COLOR_BGR2RGB)
-            video.write(pic)   #写入视频
-        video.release()
+        imageio.mimsave(video_path, pics, duration= 1000 / 20, loop=0)
 
         if mode[i] == "cadm":
             out_paths[0] = video_path
@@ -111,9 +106,9 @@ def t2m_demo():
 
     with gr.Blocks(analytics_enabled=False, css=css) as t2m_interface:
         gr.Markdown("<div align='center'> <h2> 🤷‍♂️ SemanticBoost: Elevating Motion Generation with Augmented Textual Cues </span> </h2> \
-                    <a style='font-size:18px;color: #000000' href='https://arxiv.org/abs/2211.12194'>Arxiv</a> &nbsp;&nbsp;&nbsp;&nbsp;&nbsp; \
-                    <a style='font-size:18px;color: #000000' href='https://sadtalker.github.io'>Homepage</a>  &nbsp;&nbsp;&nbsp;&nbsp;&nbsp; \
-                    <a style='font-size:18px;color: #000000' href='https://github.com/blackgold3/SemanticBoost'> Github </div>")
+                    <a style='font-size:18px;' href=''>Arxiv</a> &nbsp;&nbsp;&nbsp;&nbsp;&nbsp; \
+                    <a style='font-size:18px;' href='https://blackgold3.github.io/SemanticBoost/'>Homepage</a>  &nbsp;&nbsp;&nbsp;&nbsp;&nbsp; \
+                    <a style='font-size:18px;' href='https://github.com/blackgold3/SemanticBoost'> Github </div>")
         
         with gr.Row().style(equal_height=True):
             with gr.Column(variant='panel'): 
@@ -142,21 +137,21 @@ def t2m_demo():
                     with gr.TabItem('Results'):
                         with gr.Row():
                             with gr.Column():
-                                gen_video = gr.Video(label="CADM", format="mp4", autoplay=True, elem_classes="ov")
+                                gen_video = gr.Image(label="CADM", elem_classes="ov")
                             with gr.Column():
                                 joint_file = gr.File(label="CADM-Joints", value=None, elem_classes="dfile")
                                 smpl_file = gr.File(label="CADM-SMPL", value=None, elem_classes="dfile")
 
                         with gr.Row():
                             with gr.Column():
-                                gen_video1 = gr.Video(label="CADM-Augment", format="mp4", autoplay=True, elem_classes="ov")
+                                gen_video1 = gr.Image(label="CADM-Augment", elem_classes="ov")
                             with gr.Column():
                                 joint_file1 = gr.File(label="CADM-Augment-Joints", value=None, elem_classes="dfile")
                                 smpl_file1 = gr.File(label="CADM-Augment-SMPL", value=None, elem_classes="dfile")
                                 
                         with gr.Row():
                             with gr.Column():
-                                gen_video2 = gr.Video(label="MDM", format="mp4", autoplay=True, elem_classes="ov")
+                                gen_video2 = gr.Image(label="MDM", elem_classes="ov")
                             with gr.Column():
                                 joint_file2 = gr.File(label="MDM-Joints", value=None, elem_classes="dfile")
                                 smpl_file2 = gr.File(label="MDM-SMPL", value=None, elem_classes="dfile")