소스 검색

Add UFF test file (#511) (#593)

Lutz Roeder 4 년 전
부모
커밋
f06191366b
2개의 변경된 파일8개의 추가작업 그리고 1개의 파일을 삭제
  1. 1 1
      source/uff.js
  2. 7 0
      test/models.json

+ 1 - 1
source/uff.js

@@ -17,7 +17,7 @@ uff.ModelFactory = class {
                 tags.has(2) && tags.get(2) === 0 &&
                 tags.has(3) && tags.get(3) === 2 &&
                 tags.has(4) && tags.get(4) === 2 &&
-                tags.has(5) && tags.get(5) === 2) {
+                (!tags.has(5) || tags.get(5) === 2)) {
                 return 'uff.pb';
             }
         }

+ 7 - 0
test/models.json

@@ -6155,6 +6155,13 @@
     "format": "UFF v1",
     "link":   "https://github.com/lutzroeder/netron/issues/511"
   },
+  {
+    "type":   "uff",
+    "target": "resnet50_tensorflow_1.6.uff",
+    "source": "https://github.com/lutzroeder/netron/files/7402208/resnet50_tensorflow_1.6.uff.zip[resnet50_tensorflow_1.6.uff]",
+    "format": "UFF v1",
+    "link":   "https://github.com/lutzroeder/netron/issues/593"
+  },
   {
     "type":   "uff",
     "target": "sample_unpruned_mobilenet_v2.tar.gz",