Skip to content

Commit 8b044b7

Browse files
authored
[FEATURE] Add MoViNet template for action classification (#1742)
* Add MoViNet template for action classification Co-authored-by: Nikita Savelyev <nikita.savelyev@intel.com>
1 parent 3581917 commit 8b044b7

File tree

18 files changed

+1449
-6
lines changed

18 files changed

+1449
-6
lines changed

otx/algorithms/action/adapters/mmaction/models/__init__.py

+4-3
Original file line numberDiff line numberDiff line change
@@ -3,8 +3,9 @@
33
# Copyright (C) 2022 Intel Corporation
44
# SPDX-License-Identifier: Apache-2.0
55

6-
from .backbones import register_action_backbones
6+
from .backbones import OTXMoViNet, register_action_backbones
77
from .detectors import AVAFastRCNN
8-
from .heads import AVARoIHead
8+
from .heads import AVARoIHead, MoViNetHead
9+
from .recognizers import MoViNetRecognizer
910

10-
__all__ = ["register_action_backbones", "AVAFastRCNN", "AVARoIHead"]
11+
__all__ = ["register_action_backbones", "AVAFastRCNN", "OTXMoViNet", "MoViNetHead", "MoViNetRecognizer", "AVARoIHead"]

otx/algorithms/action/adapters/mmaction/models/backbones/__init__.py

+5
Original file line numberDiff line numberDiff line change
@@ -6,7 +6,12 @@
66
from mmaction.models.backbones.x3d import X3D
77
from mmdet.models import BACKBONES as MMDET_BACKBONES
88

9+
from .movinet import OTXMoViNet
10+
911

1012
def register_action_backbones():
1113
"""Register action backbone to mmdetection backbones."""
1214
MMDET_BACKBONES.register_module()(X3D)
15+
16+
17+
__all__ = ["OTXMoViNet"]

0 commit comments

Comments
 (0)