From 95630153c9b47e7ae4c5748c147f24633b95f3b4 Mon Sep 17 00:00:00 2001
From: cmoineau <cyril.moineau@cea.fr>
Date: Wed, 11 Sep 2024 16:02:44 +0000
Subject: [PATCH] Fix issue with weights for FC layer in HWC format.
---
aidge_export_arm_cortexm/operators.py | 18 +++++++++++++++++-
1 file changed, 17 insertions(+), 1 deletion(-)
diff --git a/aidge_export_arm_cortexm/operators.py b/aidge_export_arm_cortexm/operators.py
index 24611dd..b4e6aa8 100644
--- a/aidge_export_arm_cortexm/operators.py
+++ b/aidge_export_arm_cortexm/operators.py
@@ -51,9 +51,25 @@ class Producer_ARMCortexM(ExportNode):
def __init__(self, node, mem_info, is_input, is_output):
super().__init__(node, mem_info, is_input, is_output)
self.values = np.array(self.operator.get_output(0))
-
+ print(f"{node.name()}: {self.values.shape}")
if len(self.values.shape) == 4: # Note: export in HWC
self.values = np.transpose(self.values, (0, 2, 3, 1))
+ # The following block of code is a dirty fix for FC
+ # The issue is that FC weight in Aidge are made for an CHW input
+ # Current export is made with HWC format
+ # So we need to reorder weights of the FC
+ # Note: it is not necessary if H and W != 1 (equivalent to in_dims length == 4)
+
+ if len(self.values.shape) == 2:
+ parents = node.get_children()
+ if len(parents) == 1 and list(parents)[0].type() == "FC":
+ data_in = list(parents)[0].get_operator().get_input(0)
+ if len(data_in.dims()) == 4:
+ C = data_in.dims()[1]
+ H = data_in.dims()[2]
+ W = data_in.dims()[3]
+ # Transpose weights to adapt the HWC
+ self.values = self.values.reshape(-1, C, H, W).transpose(0, 2, 3, 1)
def export(self, export_folder: Path):
header_path = f"include/parameters/{self.attributes['name']}.hpp"
--
GitLab