jhj0517 commited on
Commit
6849ec3
1 Parent(s): ba0008b

Add sound check

Browse files
Files changed (1) hide show
  1. tests/test_config.py +9 -0
tests/test_config.py CHANGED
@@ -6,6 +6,7 @@ import functools
6
  import numpy as np
7
  import cv2
8
  from skimage.metrics import structural_similarity as compare_ssim
 
9
 
10
  from modules.utils.paths import *
11
 
@@ -70,6 +71,14 @@ def are_videos_different(video1_path: str, video2_path: str):
70
  return False
71
 
72
 
 
 
 
 
 
 
 
 
73
  @functools.lru_cache
74
  def is_cuda_available():
75
  return torch.cuda.is_available()
 
6
  import numpy as np
7
  import cv2
8
  from skimage.metrics import structural_similarity as compare_ssim
9
+ from moviepy.editor import VideoFileClip
10
 
11
  from modules.utils.paths import *
12
 
 
71
  return False
72
 
73
 
74
+ def has_sound(video_path: str):
75
+ try:
76
+ video = VideoFileClip(video_path)
77
+ return video.audio is not None
78
+ except Exception as e:
79
+ return False
80
+
81
+
82
  @functools.lru_cache
83
  def is_cuda_available():
84
  return torch.cuda.is_available()