Lutz Roeder 7 лет назад
Родитель
Сommit
cd6db6fa92
3 измененных файлов с 14 добавлено и 0 удалено
  1. 2 0
      src/tf-proto.js
  2. 8 0
      test/models.json
  3. 4 0
      test/test.js

+ 2 - 0
src/tf-proto.js

@@ -4905,6 +4905,7 @@
                         reader.value();
                         key = reader.string();
                         reader.assert("value");
+                        reader.value();
                         message.ret[key] = reader.string();
                         reader.assert("}");
                         break;
@@ -4916,6 +4917,7 @@
                         reader.value();
                         key = reader.string();
                         reader.assert("value");
+                        reader.value();
                         message.control_ret[key] = reader.string();
                         reader.assert("}");
                         break;

+ 8 - 0
test/models.json

@@ -3641,6 +3641,14 @@
     "format": "TensorFlow Saved Model v1",
     "link":   "https://github.com/onnx/tensorflow-onnx/blob/master/tests/run_pretrained_models.yaml"
   },
+  {
+    "type":   "tf",
+    "target": "resnet18_model_baseline.pbtxt",
+    "source": "https://github.com/lutzroeder/netron/files/2913372/resnet18_model_baseline.pbtxt.zip[resnet18_model_baseline.pbtxt]",
+    "render": "skip",
+    "format": "TensorFlow Graph",
+    "link":   "https://github.com/lutzroeder/netron/issues/235"
+  },
   {
     "type":   "tf",
     "target": "speech_commands_v0.pb",

+ 4 - 0
test/test.js

@@ -534,6 +534,10 @@ function next() {
         return;
     }
     var item = models.shift();
+    if (!item.type) {
+        console.error("Property 'type' is required for item '" + JSON.stringify(item) + "'.");
+        return;
+    }
     if (type && item.type != type) {
         next();
         return;