run_update_pose.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.update_anno_pose import run_update_annotation_pose_data
 
import abc
 
if __name__ == "__main__":
    args = abc.abstractproperty()
 
    #스켈레톤 업데이트된 폴더, 업데이트하면 여기에 리스트를 갱신
    args.dataset_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_231121_handsup_Act_1',
                                '/media/safemotion/HDD2/SM_Dataset/action_2022/action_cctv_img/annotations_action_v22_231121_handsup_hard_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',
                                '/media/safemotion/HDD2/SM_Dataset/action_2022/action_cctv_img/annotations_action_v22_231122_falldown_hard_Act_2',
                                '/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_2',
                                '/media/safemotion/HDD2/SM_Dataset/action_2022/action_cctv_img/annotations_action_v22_kid_hard_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_kid_kkomo_2',
                                '/media/safemotion/HDD2/SM_Dataset/action_2022/action_cctv_img/annotations_action_v22_kid_kkumjalam_1',
                                '/media/safemotion/HDD2/SM_Dataset/action_2022/action_cctv_img/annotations_action_v22_kid_kkumjalam_hard_1',
                                '/media/safemotion/HDD2/SM_Dataset/action_2022/action_cctv_img/annotations_action_v22_kid_yeongmu_1',
                                '/media/safemotion/HDD2/SM_Dataset/action_2022/action_cctv_img/annotations_action_v22_kid_yeongmu_hard_1',                                
                                '/media/safemotion/HDD2/SM_Dataset/action_2022/action_cctv_img/annotations_action_v22_mutual_act_1',
                                '/media/safemotion/HDD2/SM_Dataset/action_2022/action_cctv_img/annotations_action_v22_mutual_hard_act_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_ntu_mutual_1',
                                '/media/safemotion/HDD2/SM_Dataset/action_2022/action_cctv_img/annotations_action_v22_ntu_mutual_hard_1',
                                '/media/safemotion/HDD2/SM_Dataset/action_2022/action_cctv_img/annotations_action_v22_1st_4',
                                '/media/safemotion/HDD2/SM_Dataset/action_2022/action_cctv_img/annotations_action_v22_2nd_mutual_1',
                                '/media/safemotion/HDD2/SM_Dataset/action_2022/action_cctv_img/annotations_action_v22_3rd_1',
                                ]
    #스켈레톤 업데이트 시킬 파일의 폴더, 업데이트 완료하면 위쪽 리스트 갱신
    args.dataset_folder_list = [
                                
 
                                ]
    
    #포즈 추정 config 파일 경로
    # args.pose_config = '/workspace/smlab/smpose/configs/mmpose/hrnet_trt.py'  # noqa: E501
    args.pose_config = '/workspace/smlab/smpose/configs/mmpose/hrnet_w48_fp32_trt.py'
    
    #시각화 이미지 저장 폴더, None으로 셋팅해두면 시각화하지 않음
    args.save_pose_img_root = None #'/media/safemotion/HDD5/pjm_test/test_pose_img/'#'/workspace/pose_result/images/'
    
    #어노테이션 파일 저장 폴더, None으로 셋팅해두면 기존 파일에 스켈레톤 좌표만 업데이트됨
    args.save_json_root = None#'/media/safemotion/HDD2/SM_Dataset/action_2022/action_cctv_img/annotations_action_1st_2'
 
    #스킵할 파일 수량, 프로그램상 오류가 나서 멈추었을 경우 진행되었던 파일까진 스킵하고 진행할 수 있도록 함
    args.skip = -1
 
    #스켈레톤 업데이트 진행
    run_update_annotation_pose_data(args)