[FEATURE] Add MoViNet template for action classification (#1742)

* Add MoViNet template for action classification Co-authored-by: Nikita Savelyev <nikita.savelyev@intel.com>
openvinotoolkit · Mar 14, 2023 · 8b044b7 · 8b044b7
1 parent 3581917
commit 8b044b7
Show file tree

Hide file tree

Showing 18 changed files with 1,449 additions and 6 deletions.
diff --git a/otx/algorithms/action/adapters/mmaction/models/__init__.py b/otx/algorithms/action/adapters/mmaction/models/__init__.py
@@ -3,8 +3,9 @@
 # Copyright (C) 2022 Intel Corporation
 # SPDX-License-Identifier: Apache-2.0
 
-from .backbones import register_action_backbones
+from .backbones import OTXMoViNet, register_action_backbones
 from .detectors import AVAFastRCNN
-from .heads import AVARoIHead
+from .heads import AVARoIHead, MoViNetHead
+from .recognizers import MoViNetRecognizer
 
-__all__ = ["register_action_backbones", "AVAFastRCNN", "AVARoIHead"]
+__all__ = ["register_action_backbones", "AVAFastRCNN", "OTXMoViNet", "MoViNetHead", "MoViNetRecognizer", "AVARoIHead"]
diff --git a/otx/algorithms/action/adapters/mmaction/models/backbones/__init__.py b/otx/algorithms/action/adapters/mmaction/models/backbones/__init__.py
@@ -6,7 +6,12 @@
 from mmaction.models.backbones.x3d import X3D
 from mmdet.models import BACKBONES as MMDET_BACKBONES
 
+from .movinet import OTXMoViNet
+
 
 def register_action_backbones():
     """Register action backbone to mmdetection backbones."""
     MMDET_BACKBONES.register_module()(X3D)
+
+
+__all__ = ["OTXMoViNet"]