Ver código fonte

Update DaVinci OM detection (#799)

Lutz Roeder 4 anos atrás
pai
commit
089028056d
1 arquivos alterados com 1 adições e 1 exclusões
  1. 1 1
      source/view.js

+ 1 - 1
source/view.js

@@ -1561,7 +1561,7 @@ view.ModelFactoryService = class {
         this.register('./mlnet', [ '.zip' ]);
         this.register('./acuity', [ '.json' ]);
         this.register('./imgdnn', [ '.dnn', 'params', '.json' ]);
-        this.register('./om', [ '.om', '.onnx', '.pb' ]);
+        this.register('./om', [ '.om', '.onnx', '.pb', '.engine' ]);
     }
 
     register(id, extensions) {