main.py 1.2 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243
  1. import os
  2. import time
  3. from pathlib import Path
  4. from utils.audio_analysis import audio_analysis_pipeline
  5. from modules.video_processing.video_concat import merge_videos
  6. from modules.video_processing.video_cut import video_cut_by_json
  7. from utils.llm_director import director_json
  8. from utils.video_concat import concat_videos
  9. from utils.common import read_video_list, find_vidoe2cut
  10. from utils.first_cut import cut_once
  11. from utils.oral_cut import oral_cut
  12. from utils.aide_cut import aide_cut
  13. from utils.show_cut import show_cut
  14. from utils.logger_config import setup_logger
  15. logger = setup_logger(__name__)
  16. def main(video_path):
  17. print("start!")
  18. start = time.time()
  19. # 1、同款衣服视频裁切
  20. _, video_list = cut_once(video_path)
  21. for sub_video in video_list:
  22. # 2、口播视频裁切拼接
  23. video_2 = oral_cut(sub_video)
  24. # 3、走秀视频片段裁切 + 配乐
  25. audio_json = f"data/audio_json/{Path(sub_video).stem}.json"
  26. video_3 = show_cut(audio_json)
  27. # 4、口播走秀片段拼接
  28. concat_videos([video_2, video_3])
  29. logger.info(f"total cost {time.time() - start} sec.")
  30. if __name__ == "__main__":
  31. main("data/raw_video/rawvideo.flv")