run_gen_clip.py

Go to the documentation of this file.

import sys
sys_path = ['/workspace/smlab', '/workspace']
for path in sys_path:
    if path not in sys.path:
        sys.path.append(path)
        
from smdataset.gen_data.generate_clip import generate_clip
from smutils.utils_os import create_directory
import os
 
if __name__ == "__main__":
    save_folder = '/media/safemotion/HDD5/Inno_Test/dataset/action_2nd'
    json_folder_list = [
                        '/media/safemotion/HDD2/SM_Dataset/action_2022/action_cctv_img/annotations_action_v22_1st_1',
                        '/media/safemotion/HDD2/SM_Dataset/action_2022/action_cctv_img/annotations_action_v22_1st_2',
                        '/media/safemotion/HDD2/SM_Dataset/action_2022/action_cctv_img/annotations_action_v22_1st_3',
                        '/media/safemotion/HDD2/SM_Dataset/action_2022/action_cctv_img/annotations_action_v22_2nd_1',
                        '/media/safemotion/HDD2/SM_Dataset/action_2022/action_cctv_img/annotations_action_v22_2nd_2',
                        '/media/safemotion/HDD2/SM_Dataset/action_2022/action_cctv_img/annotations_action_v22_2nd_3',
                        '/media/safemotion/HDD2/SM_Dataset/action_2022/action_cctv_img/annotations_action_v22_2nd_4',
                        '/media/safemotion/HDD2/SM_Dataset/action_2022/action_cctv_img/annotations_action_v22_231120_Act_1',
                        '/media/safemotion/HDD2/SM_Dataset/action_2022/action_cctv_img/annotations_action_v22_kid_1',
                        '/media/safemotion/HDD2/SM_Dataset/action_2022/action_cctv_img/annotations_action_v22_kid_kkomo_1',
                        '/media/safemotion/HDD2/SM_Dataset/action_2022/action_cctv_img/annotations_action_v22_wis',
                        '/media/safemotion/HDD2/SM_Dataset/action_2022/action_cctv_img/fall_down_2021_v22',
                        '/media/safemotion/HDD2/SM_Dataset/action_2022/action_cctv_img/annotations_action_v22_231121_handsup_Act_1',
                        '/media/safemotion/HDD2/SM_Dataset/action_2022/action_cctv_img/annotations_action_v22_231122_falldown_Act_1',
                        '/media/safemotion/HDD2/SM_Dataset/action_2022/action_cctv_img/annotations_action_v22_231122_falldown_Act_2',
                        ]
    # json_folder_list = ['/media/safemotion/HDD2/SM_Dataset/action_2022/action_cctv_img/annotations_action_v22_kid_kkomo_1']
    
    cat_list={'action_upper' : 6,
              'action_lower' : 12,
              'pose' : 8, 
              'hand' : 4,
              'foot' : 2}
 
    for cat, n in cat_list.items():
        for i in range(n):
            folder = os.path.join(save_folder, cat, f'{i:02d}')
            create_directory(folder)
    
    for json_folder in json_folder_list:
        generate_clip(json_folder, save_folder)