|
@@ -270,12 +270,12 @@ def timer_check():
|
|
|
|
|
|
if __name__ == '__main__':
|
|
|
# timer_check()
|
|
|
- feature_df = getVideoInfoInXlxs('past_videos.xlsx')
|
|
|
+ feature_df = getVideoInfoInXlxs('aigc-test/past_videos.xlsx')
|
|
|
video_id_list = feature_df['videoid'].to_list()
|
|
|
video_info = {}
|
|
|
for video_id in video_id_list:
|
|
|
titleObj = feature_df[feature_df['videoid']
|
|
|
- == video_id]['title'].values[0]
|
|
|
+ == video_id]['title'].values[0]
|
|
|
video_path = feature_df[feature_df['videoid']
|
|
|
== video_id]['video_path'].values[0]
|
|
|
title = str(titleObj)
|
|
@@ -287,7 +287,7 @@ if __name__ == '__main__':
|
|
|
# print(video_id, title)
|
|
|
print(len(video_info))
|
|
|
# 获取已asr识别的视频
|
|
|
- asr_folder = 'asr_res'
|
|
|
+ asr_folder = 'aigc-test/asr_res'
|
|
|
retry = 0
|
|
|
while retry < 30:
|
|
|
asr_file_list = os.listdir(asr_folder)
|
|
@@ -297,7 +297,7 @@ if __name__ == '__main__':
|
|
|
continue
|
|
|
retry = 0
|
|
|
for asr_filename in asr_file_list:
|
|
|
- video_id = asr_filename[:-4]
|
|
|
+ video_id = int(asr_filename[:-4])
|
|
|
if video_id not in video_id_list:
|
|
|
continue
|
|
|
asr_file = os.path.join(asr_folder, asr_filename)
|