瀏覽代碼

Update ncnn.js (#833)

Lutz Roeder 4 年之前
父節點
當前提交
d2d78d82e2
共有 1 個文件被更改,包括 2 次插入1 次删除
  1. 2 1
      source/ncnn.js

+ 2 - 1
source/ncnn.js

@@ -5,6 +5,7 @@ var base = base || require('./base');
 
 // https://github.com/Tencent/ncnn/wiki/param-and-model-file-structure
 // https://github.com/Tencent/ncnn/wiki/operation-param-weight-table
+// https://github.com/Tencent/ncnn/wiki/operators
 
 ncnn.ModelFactory = class {
 
@@ -343,7 +344,7 @@ ncnn.Node = class {
                 const kernel_w = parseInt(attributes.get('1') || 0, 10);
                 const kernel_h = parseInt(attributes.get('11') || kernel_w, 10);
                 const weight_data_size = parseInt(attributes.get('6') || 0, 10);
-                this._weight(blobReader, 'weight', [ num_output, weight_data_size / ( num_output * kernel_w * kernel_h), kernel_w, kernel_h ]);
+                this._weight(blobReader, 'weight', [ num_output, weight_data_size / ( num_output * kernel_w * kernel_h), kernel_h, kernel_w ]);
                 if (parseInt(attributes.get('5') || 0, 10) === 1) {
                     this._weight(blobReader, 'bias', [ num_output ], 'float32');
                 }