run_gen_image_features.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_image_features import run_gen_image_features
 
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_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_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_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_kkomo_2',
                                '/media/safemotion/HDD2/SM_Dataset/action_2022/action_cctv_img/annotations_action_v22_kid_kkumjalam_hard_1'
                                ]
    
    #모델을 구동할 디바이스
    args.device = 'cuda:1' 
    
    #이미지 특징 저장 경로
    args.save_root = '/media/safemotion/HDD5/pjm_test/action_train_dataset_2023/image_features'
    
    #이미지 특징 생성
    run_gen_image_features(args)