From aa3edb457cc9919b0338427b07d2c496705172b0 Mon Sep 17 00:00:00 2001
From: Mayeul Aubin <mayeul.aubin@iap.fr>
Date: Wed, 25 Jun 2025 09:35:46 +0200
Subject: [PATCH] FiLM

---
 sCOCA_ML/models/FiLM.py | 16 ++++++++++++++++
 1 file changed, 16 insertions(+)
 create mode 100644 sCOCA_ML/models/FiLM.py

diff --git a/sCOCA_ML/models/FiLM.py b/sCOCA_ML/models/FiLM.py
new file mode 100644
index 0000000..e7056b5
--- /dev/null
+++ b/sCOCA_ML/models/FiLM.py
@@ -0,0 +1,16 @@
+"""
+FiLM: Feature-wise Linear Modulation
+For applying FiLM to condition feature maps on style parameters.
+"""
+import torch.nn as nn
+
+class FiLM(nn.Module):
+    def __init__(self, num_features, style_dim):
+        super(FiLM, self).__init__()
+        self.gamma = nn.Linear(style_dim, num_features)
+        self.beta = nn.Linear(style_dim, num_features)
+
+    def forward(self, x, style):
+        gamma = self.gamma(style).unsqueeze(-1).unsqueeze(-1).unsqueeze(-1)
+        beta = self.beta(style).unsqueeze(-1).unsqueeze(-1).unsqueeze(-1)
+        return gamma * x + beta