Update run_video_ccip.py

run_video_ccip.py

@@ -43,7 +43,8 @@ all_paths_txt = all_paths_mp4.map(lambda x: x.replace(".mp4", ".txt")).map(lambd
 os.makedirs("tgt_dir", exist_ok=True)
 for ele in all_paths_mp4.values.tolist() + all_paths_txt.values.tolist():
     copy2(ele, os.path.join("tgt_dir", ele.split("/")[-1]))
-    
+
+python run_video_ccip.py Beyond_the_Boundary_Videos Beyond_the_Boundary_Videos_named --image_dir named_image_dir
 '''
 
 import os
@@ -61,7 +62,7 @@ def load_images_from_directory(image_dir):
     键为图片的文件名（不含扩展名），值为图片的 PIL.Image 对象。
     """
     name_image_dict = {}
-    image_paths = list(pathlib.Path(image_dir).rglob("*.png")) + list(pathlib.Path(image_dir).rglob("*.jpg")) + list(pathlib.Path(image_dir).rglob("*.jpeg"))
+    image_paths = list(pathlib.Path(image_dir).rglob("*.png")) + list(pathlib.Path(image_dir).rglob("*.jpg")) + list(pathlib.Path(image_dir).rglob("*.jpeg")) + list(pathlib.Path(image_dir).rglob("*.webp"))
     
     for image_path in tqdm(image_paths, desc="Loading images"):
         image = Image.open(image_path)