runner_builder Namespace Reference

Functions
	build_action_runner (args)
	build_detect_runner (args)
	build_pose_runner (args)
	build_track_runner (args)

Function Documentation

build_action_runner()

runner_builder.build_action_runner

(

args

)

action/runner 의 모델을 빌드하는 함수
action/runner의 모델은 action/models의 모듈 및 추가적인 파이토치 모듈을 이용해서 정의됨

args : 
    type -> 모델의 타입  __runner_builders__ 의 키값이 들어가야함
    type을 제외한 나머지 파라미터는 모델의 파라미터값이 있어야함
        세부적으로 모듈의 타입과 모듈의 파라미터가 정의되어 있어야함

returns:
    파이토이 모델

Definition at line 9 of file runner_builder.py.

def build_action_runner(args):
    """
    action/runner 의 모델을 빌드하는 함수
    action/runner의 모델은 action/models의 모듈 및 추가적인 파이토치 모듈을 이용해서 정의됨
 
    args : 
        type -> 모델의 타입  __runner_builders__ 의 키값이 들어가야함
        type을 제외한 나머지 파라미터는 모델의 파라미터값이 있어야함
            세부적으로 모듈의 타입과 모듈의 파라미터가 정의되어 있어야함
    
    returns:
        파이토이 모델
    """
    assert args.type in __runner_builders__, \
            f'not found runner model type : {args.type}'
    
    runner_builder = __runner_builders__[args.type]
    runner_args = args.copy()
    runner_args.pop('type')
    return runner_builder(**runner_args)

build_detect_runner()

runner_builder.build_detect_runner

(

args

)

detect/runner 의 모델을 빌드하는 함수
현재 detect/runner의 모델은 오픈소스기반으로 구현되어 있음

args : 
    type -> 모델의 타입  __runner_builders__ 의 키값이 들어가야함
    type을 제외한 나머지 파라미터는 모델의 파라미터값이 있어야함
        세부적으로 모듈의 타입과 모듈의 파라미터가 정의되어 있어야함

returns:
    파이토이 모델 or TensorRT 모델

Definition at line 11 of file runner_builder.py.

def build_detect_runner(args):
    """
    detect/runner 의 모델을 빌드하는 함수
    현재 detect/runner의 모델은 오픈소스기반으로 구현되어 있음
 
    args : 
        type -> 모델의 타입  __runner_builders__ 의 키값이 들어가야함
        type을 제외한 나머지 파라미터는 모델의 파라미터값이 있어야함
            세부적으로 모듈의 타입과 모듈의 파라미터가 정의되어 있어야함
    
    returns:
        파이토이 모델 or TensorRT 모델
    """
    assert args.type in __runner_builders__, \
            f'not found detection model type : {args.type}'
    
    runner_builder = __runner_builders__[args.type]
    runner_args = args.copy()
    runner_args.pop('type')
    return runner_builder(**runner_args)

build_pose_runner()

runner_builder.build_pose_runner

(

args

)

pose/runner 의 모델을 빌드하는 함수
현재 pose/runner의 모델은 mmpose기반으로 구현되어 있음
TODO: pytorch 모델 빌드하는 기능 추가 필요

args : 
    type -> 모델의 타입  __runner_builders__ 의 키값이 들어가야함
    type을 제외한 나머지 파라미터는 모델의 파라미터값이 있어야함
        세부적으로 모듈의 타입과 모듈의 파라미터가 정의되어 있어야함

returns:
    파이토이 모델 or TensorRT 모델

Definition at line 8 of file runner_builder.py.

def build_pose_runner(args):
    """
    pose/runner 의 모델을 빌드하는 함수
    현재 pose/runner의 모델은 mmpose기반으로 구현되어 있음
    TODO: pytorch 모델 빌드하는 기능 추가 필요
 
    args : 
        type -> 모델의 타입  __runner_builders__ 의 키값이 들어가야함
        type을 제외한 나머지 파라미터는 모델의 파라미터값이 있어야함
            세부적으로 모듈의 타입과 모듈의 파라미터가 정의되어 있어야함
    
    returns:
        파이토이 모델 or TensorRT 모델
    """
    assert args.type in __runner_builders__, \
            f'not found pose model type : {args.type}'
    
    runner_builder = __runner_builders__[args.type]
    runner_args = args.copy()
    runner_args.pop('type')
    return runner_builder(**runner_args)

build_track_runner()

runner_builder.build_track_runner

(

args

)

Definition at line 7 of file runner_builder.py.

def build_track_runner(args):
    assert args.type in __runner_builders__, \
            f'not found track model type : {args.type}'
    
    runner_builder = __runner_builders__[args.type]
    runner_args = args.copy()
    runner_args.pop('type')
    return runner_builder(**runner_args)