run_gen_pkl_v22.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_pkl_data import run_generate_pkl_data_from_json_v22
from smutils.utils_os import create_directory
from smutils.utils_data import split_train_and_val_v22
 
import abc
 
if __name__ == "__main__":
    
    # category_k={'action_upper' : 60,
    #             'action_lower' : 60,
    #             'pose' : 10, 
    #             'hand' : 10,
    #             'foot' : 10}
    # category_info={'action_upper' : 6,
    #                'action_lower' : 12,
    #                'pose' : 8, 
    #                'hand' : 2,
    #                'foot' : 2}
    category_k={
                'pose' : 10, 
                'hand' : 10,
                'foot' : 10}
    category_info={
                   'pose' : 8, 
                   'hand' : 4,
                   'foot' : 2}
 
    args = abc.abstractproperty()
    args.category_k = category_k
    args.data_root = '/media/safemotion/HDD5/Inno_Test/dataset/action_2nd'
    args.pkl_root = '/media/safemotion/HDD5/pjm_test/action_train_dataset_2023/action_v22_2nd'
    
    run_generate_pkl_data_from_json_v22(args)
 
    split_data_folder = '/media/safemotion/HDD5/pjm_test/action_train_dataset_2023/action_v22_2nd_split'
    split_train_and_val_v22(args.pkl_root, split_data_folder, category_info, total_data_num=100, ratio=0.8)