private static void printSignature()

in tensorflow-examples-legacy/object_detection/src/main/java/DetectObjects.java [97:120]

• 24 lines of code
• 3 McCabe index (conditional complexity)

  private static void printSignature(SavedModelBundle model) throws Exception {
    MetaGraphDef m = MetaGraphDef.parseFrom(model.metaGraphDef());
    SignatureDef sig = m.getSignatureDefOrThrow("serving_default");
    int numInputs = sig.getInputsCount();
    int i = 1;
    System.out.println("MODEL SIGNATURE");
    System.out.println("Inputs:");
    for (Map.Entry<String, TensorInfo> entry : sig.getInputsMap().entrySet()) {
      TensorInfo t = entry.getValue();
      System.out.printf(
          "%d of %d: %-20s (Node name in graph: %-20s, type: %s)\n",
          i++, numInputs, entry.getKey(), t.getName(), t.getDtype());
    }
    int numOutputs = sig.getOutputsCount();
    i = 1;
    System.out.println("Outputs:");
    for (Map.Entry<String, TensorInfo> entry : sig.getOutputsMap().entrySet()) {
      TensorInfo t = entry.getValue();
      System.out.printf(
          "%d of %d: %-20s (Node name in graph: %-20s, type: %s)\n",
          i++, numOutputs, entry.getKey(), t.getName(), t.getDtype());
    }
    System.out.println("-----------------------------------------------");
  }