Browse Source

Update Darknet detection

Lutz Roeder 5 years ago
parent
commit
43c953f044
1 changed files with 5 additions and 3 deletions
  1. 5 3
      source/darknet.js

+ 5 - 3
source/darknet.js

@@ -6,9 +6,8 @@ var base = base || require('./base');
 darknet.ModelFactory = class {
 
     match(context) {
-        const extension = context.identifier.split('.').pop().toLowerCase();
-        if (extension == 'cfg' || extension == 'model') {
-            const reader = base.TextReader.create(context.buffer, 2048);
+        try {
+            const reader = base.TextReader.create(context.buffer);
             for (;;) {
                 const line = reader.read();
                 if (line === undefined) {
@@ -23,6 +22,9 @@ darknet.ModelFactory = class {
                 }
             }
         }
+        catch (err) {
+            // continue regardless of error
+        }
         return false;
     }