brian/visual_critical_discourse_analysis
1
0
Fork 0
Code Issues 11 Pull Requests Actions Projects 1 Releases Activity

changed returned class type

Browse Source
This commit is contained in:
brian
2025-03-18 19:28:12 +00:00
parent a379270add
commit 7d8af1c582
1 changed files with 2 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files
model/src/models/visual_communication.py
+2 -6
View File
@@ -4,8 +4,6 @@ from __future__ import annotations
from torch import nn from torch import nn
from shared.docstore.src.classes import ModelData
from .angle import AngleTail from .angle import AngleTail
from .contact import ContactTail from .contact import ContactTail
from .distance import DistanceTail from .distance import DistanceTail
@@ -39,7 +37,7 @@ class VisualCommunicationModel(nn.Module):
self.framing_tail = FramingTail() self.framing_tail = FramingTail()
self.salience_tail = SalienceTail() self.salience_tail = SalienceTail()
def forward(self, x) -> ModelData: def forward(self, x) -> dict:
"""Calculate model output on data.""" """Calculate model output on data."""
# generate visual representation # generate visual representation
features = self.resnet_head(x) features = self.resnet_head(x)
@@ -57,6 +55,4 @@ class VisualCommunicationModel(nn.Module):
'framing': self.framing_tail(features), 'framing': self.framing_tail(features),
'salience': self.salience_tail(features), 'salience': self.salience_tail(features),
} }
# convert to respective classes return prediction_dict
data = ModelData.from_prediction_dict(prediction_dict)
return data