2
0
Эх сурвалжийг харах

Update TensorFlow Lite detection

Lutz Roeder 6 жил өмнө
parent
commit
5312521619
1 өөрчлөгдсөн 7 нэмэгдсэн , 17 устгасан
  1. 7 17
      src/tflite.js

+ 7 - 17
src/tflite.js

@@ -22,25 +22,15 @@ tflite.ModelFactory = class {
 
     open(context, host) {
         return host.require('./tflite-schema').then((tflite_schema) => {
-            const identifier = context.identifier;
-            let model = null;
-            try {
-                const buffer = context.buffer;
-                const byteBuffer = new flatbuffers.ByteBuffer(buffer);
-                tflite.schema = tflite_schema;
-                if (!tflite.schema.Model.bufferHasIdentifier(byteBuffer)) {
-                    throw new tflite.Error("File format is not tflite.Model.");
-                }
-                model = tflite.schema.Model.getRootAsModel(byteBuffer);
-            }
-            catch (error) {
-                host.exception(error, false);
-                const message = error && error.message ? error.message : error.toString();
-                throw new tflite.Error(message.replace(/\.$/, '') + " in '" + identifier + "'.");
-            }
-
             return tflite.Metadata.open(host).then((metadata) => {
+                const identifier = context.identifier;
                 try {
+                    const buffer = new flatbuffers.ByteBuffer(context.buffer);
+                    tflite.schema = tflite_schema;
+                    if (!tflite.schema.Model.bufferHasIdentifier(buffer)) {
+                        throw new tflite.Error("File format is not tflite.Model.");
+                    }
+                    const model = tflite.schema.Model.getRootAsModel(buffer);
                     return new tflite.Model(metadata, model);
                 }
                 catch (error) {