获取 1 , 2月 label

functions/generate_data.py

@@ -137,16 +137,18 @@ class VideoDataGenerator(object):
 
 
 if __name__ == '__main__':
-    date_list = generate_hourly_strings("2024010100", "2024013123")
+    # date_list = generate_hourly_strings("2024010100", "2024013123")
+    date_list = generate_daily_strings("20240101", "20240228")
     V = VideoDataGenerator()
     L = {}
     # print(date_list)
     for date_str in tqdm(date_list):
         L[date_str] = {}
-        data_list = V.get_hour_data(date_str)
+        # data_list = V.get_hour_data(date_str)
+        data_list = V.get_daily_data(date_str)
         for obj in tqdm(data_list):
             video_id = obj['video_id']
             L[date_str][video_id] = obj
 
-    with open('data/january_train.json', 'w') as f:
+    with open('data/jan_feb_label.json', 'w') as f:
         f.write(json.dumps(L, ensure_ascii=False, indent=4))