vidaio / test.py
DavidHstar's picture
Update model files
f72237c verified
import os
if __name__ == "__main__":
video_dir = "../test_video"
video_size = {}
for video_file in os.listdir(video_dir):
if video_file.endswith(".mp4"):
size = os.path.getsize(os.path.join(video_dir, video_file))
video_size[video_file] = size
sorted_videos = sorted(video_size.items(), key=lambda x: x[1], reverse=True)
duplicate_videos = []
seen_sizes = set()
for video, size in sorted_videos:
if size in seen_sizes:
duplicate_videos.append(video)
else:
seen_sizes.add(size)
print("Duplicate videos based on size:")
for idx, video in enumerate(duplicate_videos):
# if idx == 0: continue
print(video)
os.remove(os.path.join(video_dir, video))
# if __name__ == "__main__":
# video_dir1 = "../test_video1"
# video_dir2 = "../test_video2"
# video_name1 = []
# for video_file in os.listdir(video_dir1):
# if video_file.endswith(".mp4"):
# video_name1.append(video_file)
# video_name2 = []
# for video_file in os.listdir(video_dir2):
# if video_file.endswith(".mp4"):
# video_name2.append(video_file)
# duplicate_videos = []
# for video in video_name1:
# if video in video_name2:
# duplicate_videos.append(video)
# print("Duplicate videos based on name:")
# for idx, video in enumerate(duplicate_videos):
# # if idx == 0: continue
# print(video)
# os.remove(os.path.join(video_dir1, video))