Lutz Roeder 5 лет назад
Родитель
Сommit
2bed4ce7db
1 измененных файлов с 8 добавлено и 7 удалено
  1. 8 7
      source/armnn.js

+ 8 - 7
source/armnn.js

@@ -6,15 +6,16 @@ var flatbuffers = flatbuffers || require('./flatbuffers');
 armnn.ModelFactory = class {
 
     match(context) {
-        const extension = context.identifier.split('.').pop().toLowerCase();
-        if (extension == 'armnn') {
-            return true;
-        }
-        if (extension === 'json') {
-            const tags = context.tags('json');
-            if (tags.has('layers') && tags.has('inputIds') && tags.has('outputIds')) {
+        switch (context.identifier.split('.').pop().toLowerCase()) {
+            case 'armnn': {
                 return true;
             }
+            case 'json': {
+                const tags = context.tags('json');
+                if (tags.has('layers') && tags.has('inputIds') && tags.has('outputIds')) {
+                    return true;
+                }
+            }
         }
         return false;
     }