sunxy 11 月之前
父節點
當前提交
8ad00d9a03
共有 1 個文件被更改,包括 4 次插入4 次删除
  1. 4 4
      ai_tag_task.py

+ 4 - 4
ai_tag_task.py

@@ -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)