mmaction_base_runner.py

Go to the documentation of this file.

from abc import abstractmethod
 
from mmengine.dataset import Compose
from mmengine.registry import init_default_scope
from mmaction.apis import init_recognizer
 
 
class MMActionBaseRunner:
 
    def __init__(self, model_cfg, model_checkpoint, device):
 
        self.model = init_recognizer(model_cfg, model_checkpoint, device)
 
        cfg = self.model.cfg
        init_default_scope(cfg.get('default_scope', 'mmaction'))
        test_pipeline_cfg = cfg.test_pipeline
        self.test_pipeline = Compose(test_pipeline_cfg)
 
    @abstractmethod
    def make_action_input(self, *args, **kwargs ):
        pass
 
    @abstractmethod
    def run_recognizer(self, *args, **kwargs):
        pass