From ccce8ba20c2c7a8f906d357daad2badd60f8e0de Mon Sep 17 00:00:00 2001 From: Axel Farrugia Date: Thu, 27 Jun 2024 14:47:01 +0200 Subject: [PATCH] Unfix multi outputs --- python/n2d2/deepnet.py | 6 ++++++ 1 file changed, 6 insertions(+) diff --git a/python/n2d2/deepnet.py b/python/n2d2/deepnet.py index 773e0274..fbf777a6 100755 --- a/python/n2d2/deepnet.py +++ b/python/n2d2/deepnet.py @@ -111,10 +111,16 @@ def get_output_cells(self): Return the last N2D2 cells in the deepNet """ output = [] + """ + // Made to support several outputs but breaks regular networks exports for cell in self.N2D2().getCells(): if len(self._cells[cell].N2D2().getChildrenCells()) == 0: output.append(self._cells[cell]) return output + """ + for cell in self.N2D2().getLayers()[-1]: + output.append(self._cells[cell]) + return output def draw(self, filename): N2D2.DrawNet.draw(self._N2D2_object, filename)