Skip to content

Develop nomenclature update #29

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Closed
rowem5 wants to merge 5 commits into from
Closed

Develop nomenclature update #29

Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
5 commits
Select commit Hold shift + click to select a range
0cda9c0
Update video_datamodule.py
rowem5 Jan 17, 2026
e52f5d4
Update __init__.py
rowem5 Jan 17, 2026
9b1ebe5
Create __init__.py
rowem5 Jan 17, 2026
0af382b
Update __init__.py
rowem5 Jan 17, 2026
f159d73
Update video_segmenter.py
rowem5 Jan 17, 2026
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
1 change: 0 additions & 1 deletion lightning_action/data/video_datamodule.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -810,4 +810,3 @@ def input_size(self) -> int:
def num_classes(self) -> int:
"""Get the number of action classes."""
return self.dataset.num_classes
return self.dataset.num_classes
5 changes: 3 additions & 2 deletions lightning_action/models/__init__.py
View file
Open in desktop
Original file line number Diff line number Diff line change
@@ -1,7 +1,7 @@
"""Models for action segmentation."""

from .heads import RNN, DilatedTCN, TemporalMLP
from .segmenter import BaseModel, Segmenter
from lightning_action.models.heads import RNN, DilatedTCN, TemporalMLP
from lightning_action.models.segmenter import BaseModel, Segmenter

__all__ = [
'BaseModel',
Expand All @@ -10,3 +10,4 @@
'RNN',
'TemporalMLP',
]

11 changes: 11 additions & 0 deletions lightning_action/models/backbones/__init__.py
View file
Open in desktop
Original file line number Diff line number Diff line change
@@ -0,0 +1,11 @@
"""Backbone architectures for action segmentation models."""

from lightning_action.models.backbones.resnet import ResNetBackbone
from lightning_action.models.backbones.resnet_beast import ResNetBeastBackbone
from lightning_action.models.backbones.vitmae import ViTMAEBackbone

__all__ = [
'ResNetBackbone',
'ResNetBeastBackbone',
'ViTMAEBackbone',
]
8 changes: 5 additions & 3 deletions lightning_action/models/heads/__init__.py
View file
Open in desktop
Original file line number Diff line number Diff line change
@@ -1,11 +1,13 @@
"""Head architectures for action segmentation models."""

from .rnn import RNN
from .tcn import DilatedTCN
from .temporalmlp import TemporalMLP
from lightning_action.models.heads.rnn import RNN
from lightning_action.models.heads.tcn import DilatedTCN
from lightning_action.models.heads.temporalmlp import TemporalMLP
from lightning_action.models.necks.mha_pooling import MultiheadAttentionPooling

__all__ = [
'DilatedTCN',
'RNN',
'TemporalMLP',
'MultiheadAttentionPooling'
]
8 changes: 5 additions & 3 deletions lightning_action/models/video_segmenter.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -20,9 +20,11 @@
from typeguard import typechecked

from lightning_action.models.segmenter import BaseModel
from lightning_action.models.backbones.vitmae import ViTMAEBackbone
from lightning_action.models.backbones.resnet import ResNetBackbone
from lightning_action.models.backbones.resnet_beast import ResNetBeastBackbone
from lightning_action.models.backbones import (
ResNetBackbone,
ResNetBeastBackbone,
ViTMAEBackbone,
)
from lightning_action.models.necks.mha_pooling import MultiheadAttentionPooling
from lightning_action.models.heads import DilatedTCN, TemporalMLP, RNN

Expand Down
Loading