From f1392067586390d286e6104025b54f827adf957c Mon Sep 17 00:00:00 2001 From: Brian Bjarke Jensen Date: Tue, 12 Mar 2024 21:50:47 +0100 Subject: [PATCH] added model definitions --- model/src/models/__init__.py | 0 model/src/models/angle.py | 8 ++++++ model/src/models/contact.py | 8 ++++++ model/src/models/distance.py | 8 ++++++ model/src/models/framing.py | 8 ++++++ model/src/models/fully_connected.py | 26 +++++++++++++++++++ model/src/models/information_value.py | 8 ++++++ model/src/models/modality_color.py | 8 ++++++ model/src/models/modality_depth.py | 8 ++++++ model/src/models/modality_lighting.py | 8 ++++++ model/src/models/point_of_view.py | 8 ++++++ .../src/{model.py => models/resnet18_head.py} | 0 model/src/models/salience.py | 8 ++++++ model/src/models/visual_syntax.py | 8 ++++++ 14 files changed, 114 insertions(+) create mode 100644 model/src/models/__init__.py create mode 100644 model/src/models/angle.py create mode 100644 model/src/models/contact.py create mode 100644 model/src/models/distance.py create mode 100644 model/src/models/framing.py create mode 100644 model/src/models/fully_connected.py create mode 100644 model/src/models/information_value.py create mode 100644 model/src/models/modality_color.py create mode 100644 model/src/models/modality_depth.py create mode 100644 model/src/models/modality_lighting.py create mode 100644 model/src/models/point_of_view.py rename model/src/{model.py => models/resnet18_head.py} (100%) create mode 100644 model/src/models/salience.py create mode 100644 model/src/models/visual_syntax.py diff --git a/model/src/models/__init__.py b/model/src/models/__init__.py new file mode 100644 index 0000000..e69de29 diff --git a/model/src/models/angle.py b/model/src/models/angle.py new file mode 100644 index 0000000..2fde0ad --- /dev/null +++ b/model/src/models/angle.py @@ -0,0 +1,8 @@ +from __future__ import annotations + +from .fully_connected import FullyConnectedModel + + +class AngleTail(FullyConnectedModel): + def __init__(self): + super().__init__(num_out_features=3) diff --git a/model/src/models/contact.py b/model/src/models/contact.py new file mode 100644 index 0000000..be69b23 --- /dev/null +++ b/model/src/models/contact.py @@ -0,0 +1,8 @@ +from __future__ import annotations + +from .fully_connected import FullyConnectedModel + + +class ContactTail(FullyConnectedModel): + def __init__(self): + super().__init__(num_out_features=2) diff --git a/model/src/models/distance.py b/model/src/models/distance.py new file mode 100644 index 0000000..d5a7d8c --- /dev/null +++ b/model/src/models/distance.py @@ -0,0 +1,8 @@ +from __future__ import annotations + +from .fully_connected import FullyConnectedModel + + +class DistanceTail(FullyConnectedModel): + def __init__(self): + super().__init__(num_out_features=3) diff --git a/model/src/models/framing.py b/model/src/models/framing.py new file mode 100644 index 0000000..25a6c10 --- /dev/null +++ b/model/src/models/framing.py @@ -0,0 +1,8 @@ +from __future__ import annotations + +from .fully_connected import FullyConnectedModel + + +class FramingTail(FullyConnectedModel): + def __init__(self): + super().__init__(num_out_features=4) diff --git a/model/src/models/fully_connected.py b/model/src/models/fully_connected.py new file mode 100644 index 0000000..a5bb915 --- /dev/null +++ b/model/src/models/fully_connected.py @@ -0,0 +1,26 @@ +from __future__ import annotations + +import torch.nn as nn + + +class FullyConnectedModel(nn.Module): + def __init__(self, num_out_features: int): + super().__init__() + # define layers + self.fc1 = nn.Linear(in_features=16*16*512, out_features=512) + self.af1 = nn.ReLU() + self.fc2 = nn.Linear(in_features=512, out_features=128) + self.af2 = nn.ReLU() + self.fc3 = nn.Linear(in_features=128, out_features=32) + self.af3 = nn.ReLU() + self.fc4 = nn.Linear(in_features=32, out_features=num_out_features) + + def forward(self, x): + x = self.fc1(x) + x = self.af1(x) + x = self.fc2(x) + x = self.af2(x) + x = self.fc3(x) + x = self.af3(x) + x = self.fc4(x) + return x diff --git a/model/src/models/information_value.py b/model/src/models/information_value.py new file mode 100644 index 0000000..31c2848 --- /dev/null +++ b/model/src/models/information_value.py @@ -0,0 +1,8 @@ +from __future__ import annotations + +from .fully_connected import FullyConnectedModel + + +class InformationValueTail(FullyConnectedModel): + def __init__(self): + super().__init__(num_out_features=3) diff --git a/model/src/models/modality_color.py b/model/src/models/modality_color.py new file mode 100644 index 0000000..ae10bd8 --- /dev/null +++ b/model/src/models/modality_color.py @@ -0,0 +1,8 @@ +from __future__ import annotations + +from .fully_connected import FullyConnectedModel + + +class ModalityColorTail(FullyConnectedModel): + def __init__(self): + super().__init__(num_out_features=3) diff --git a/model/src/models/modality_depth.py b/model/src/models/modality_depth.py new file mode 100644 index 0000000..3c0e5e8 --- /dev/null +++ b/model/src/models/modality_depth.py @@ -0,0 +1,8 @@ +from __future__ import annotations + +from .fully_connected import FullyConnectedModel + + +class ModalityDepthTail(FullyConnectedModel): + def __init__(self): + super().__init__(num_out_features=3) diff --git a/model/src/models/modality_lighting.py b/model/src/models/modality_lighting.py new file mode 100644 index 0000000..8913281 --- /dev/null +++ b/model/src/models/modality_lighting.py @@ -0,0 +1,8 @@ +from __future__ import annotations + +from .fully_connected import FullyConnectedModel + + +class ModalityLightingTail(FullyConnectedModel): + def __init__(self): + super().__init__(num_out_features=3) diff --git a/model/src/models/point_of_view.py b/model/src/models/point_of_view.py new file mode 100644 index 0000000..d0d8ae3 --- /dev/null +++ b/model/src/models/point_of_view.py @@ -0,0 +1,8 @@ +from __future__ import annotations + +from .fully_connected import FullyConnectedModel + + +class PointOfViewTail(FullyConnectedModel): + def __init__(self): + super().__init__(num_out_features=2) diff --git a/model/src/model.py b/model/src/models/resnet18_head.py similarity index 100% rename from model/src/model.py rename to model/src/models/resnet18_head.py diff --git a/model/src/models/salience.py b/model/src/models/salience.py new file mode 100644 index 0000000..bc9f0af --- /dev/null +++ b/model/src/models/salience.py @@ -0,0 +1,8 @@ +from __future__ import annotations + +from .fully_connected import FullyConnectedModel + + +class SalienceTail(FullyConnectedModel): + def __init__(self): + super().__init__(num_out_features=5) diff --git a/model/src/models/visual_syntax.py b/model/src/models/visual_syntax.py new file mode 100644 index 0000000..fe02a2b --- /dev/null +++ b/model/src/models/visual_syntax.py @@ -0,0 +1,8 @@ +from __future__ import annotations + +from .fully_connected import FullyConnectedModel + + +class VisualSyntaxTail(FullyConnectedModel): + def __init__(self): + super().__init__(num_out_features=18)