Viewer - add layer indices to graph output

[tidl/tidl-api.git] / viewer / dot_graph.cpp

diff --git a/viewer/dot_graph.cpp b/viewer/dot_graph.cpp
index 25b6a97550dd6194d1e3eff0500db88659f8a36e..e281c195c186a281421e822d30fcf167a6ad63d2 100644 (file)
--- a/viewer/dot_graph.cpp
+++ b/viewer/dot_graph.cpp
@@ -102,17 +102,19 @@ void DotGraph::AddVertices()
 
         auto V = add_vertex(i, *sub);
 
-        AddVertexProperties(V, sub, layer);
+        AddVertexProperties(V, sub, layer, i);
     }
 
 }
 
 void DotGraph::AddVertexProperties(Vertex& V, Graph* g,
-                                   const sTIDL_Layer_t& layer)
+                                   const sTIDL_Layer_t& layer,
+                                   int index)
 {
     VertexPropertyMap<Graph> vpm = boost::get(vertex_attribute, *g);
     vpm[V][COLOR] = GetVertexColor(layer.layerType);
     vpm[V][STYLE] = BOLD;
+    vpm[V]["xlabel"] = std::to_string(index);
     vpm[V][LABEL] = "{";
 
     switch (layer.layerType)
@@ -121,7 +123,7 @@ void DotGraph::AddVertexProperties(Vertex& V, Graph* g,
         {
             vpm[V][LABEL] += TIDL_LayerString[layer.layerType];
             const sTIDL_ConvParams_t& p = layer.layerParams.convParams;
-            vpm[V][LABEL] += "\\n" + std::to_string(p.kernelW) + "x" +
+            vpm[V][LABEL] += " " + std::to_string(p.kernelW) + "x" +
                              std::to_string(p.kernelH);
 
             if (p.enablePooling)