run_gen_clip_v22_mtml.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_mtml import generate_clip
from smutils.utils_os import create_directory
import os
 
if __name__ == "__main__":
    #클립 저장 폴더
    save_folder = '/media/safemotion/HDD5/pjm_test/action_train_dataset_2023/action_2024'
 
    #어노테이션 파일 저장 폴더 리스트
    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',
                        ]
    # 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)