runner_builder.py

Go to the documentation of this file.

from smaction.runner.mmaction_pytorch_runner import MMActionPyTorchRunner
from smaction.runner.action_recognition_runner import ActionRecognitionRunner
 
__runner_builders__ = {
    "MMActionPyTorchRunner": MMActionPyTorchRunner,
    "ActionRecognitionRunner" : ActionRecognitionRunner,
}
 
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)