run_gen_action_pkl.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, run_merge_pkl, merge_pkl
from smutils.utils_data import split_train_and_val, load_pkl_data, print_data_num_per_class, split_train_and_val, dataset_class_filtering
from smdataset.gen_data.cvt_labelmap import *
from smutils.utils_os import create_directory
 
import mmcv
import abc
import argparse
 
 
parser = argparse.ArgumentParser(description="mode select")
parser.add_argument("-mode", type=int, default=0)
 
 
 
def splite_clip():
    
    pass
 
if __name__ == "__main__":
    args = parser.parse_args()
    print(args.mode)
    
    base_name = 'action_add_1st_1'
    name_tag = 'k60'
    
    args = abc.abstractproperty()
    # args.dataset_folder = '/media/safemotion/HDD2/SM_Dataset/action_2022/action_cctv_img/annotations_action_test6'
    # args.dataset_folder = '/media/safemotion/HDD2/SM_Dataset/action_2022/action_cctv_img/annotations_action_wis'
    args.dataset_folder = '/media/safemotion/HDD2/SM_Dataset/action_2022/action_cctv_img/annotations_action_1st_1'
    args.pkl_root = f'/media/safemotion/HDD5/pjm_test/action_mc_2023_pkl/{base_name}_{name_tag}/'
 
    merge_folder = '/media/safemotion/HDD5/pjm_test/action_mc_2023_merge'
    create_directory(merge_folder)
    args.merge_pkl_path = f'{merge_folder}/{base_name}_{name_tag}_merge.pkl'
    args.mode = 'mc'
    args.k = 60
 
    run_generate_pkl_data_from_json(args)
    run_merge_pkl(args)
    
 
 
    # args.pkl_2021_train_ori = '/workspace/pkl_dataset/action_2021/smaction_2021_common_2nd_train.pkl'
    # args.pkl_2021_test_ori = '/workspace/pkl_dataset/action_2021/smaction_2021_common_2nd_test.pkl'
 
    # args.pkl_2021_train = '/workspace/pkl_dataset/action_2021_cvt_c17/smaction_2021_common_2nd_train.pkl'
    # args.pkl_2021_test = '/workspace/pkl_dataset/action_2021_cvt_c17/smaction_2021_common_2nd_test.pkl'
    # args.train_save_path = f'/workspace/pkl_dataset/action_2022/{base_name}_{name_tag}_train.pkl'
    # args.val_save_path = f'/workspace/pkl_dataset/action_2022/{base_name}_{name_tag}_val.pkl'
    # args.k = 75
    # args.c = 13
    # args.mode = 'val'
 
    # # run_generate_pkl_data_from_json(args)
    # # run_merge_pkl(args)
 
    # # print_data_num_per_class(args.merge_pkl_path, simple_lablemap)
 
    # pkl_path_list = [
    #     '/workspace/pkl_dataset/action_2022_merge/action_wis_k75_filtered_merge.pkl',
    #     '/workspace/pkl_dataset/action_2022_merge/action_kid_k75_merge.pkl'
    # ]
    # merged_pkl_datas = merge_pkl(folder=None, file_path_list=pkl_path_list)
 
    
    # class_num = args.c
    # max_train_data_num = 1900
    # max_val_data_num = 1900
    # train_list, val_list = split_train_and_val(merged_pkl_datas, class_num, max_train_data_num, max_val_data_num)
 
    # print('train')
    # print_data_num_per_class(train_list, simple_lablemap)
    # print('val')
    # print_data_num_per_class(val_list, simple_lablemap)
 
    # mmcv.dump(train_list, args.train_save_path)    
    # mmcv.dump(val_list, args.val_save_path)
 
 
 
    # save_pkl_path = f'/workspace/pkl_dataset/action_2022_merge/{base_name}_{name_tag}_filtered_merge.pkl'
 
    # pkl_datas = dataset_class_filtering(args.merge_pkl_path, [9, 10, 11, 12])
    # mmcv.dump(pkl_datas, save_pkl_path)
 
    # print_data_num_per_class(merged_pkl_datas, simple_lablemap)