Ver código fonte

Update kmodel.js

Lutz Roeder 3 anos atrás
pai
commit
8d45c0d799
1 arquivos alterados com 18 adições e 0 exclusões
  1. 18 0
      source/kmodel.js

+ 18 - 0
source/kmodel.js

@@ -588,6 +588,24 @@ kmodel.Reader = class {
                         layer.inputs[0].arguments[0].shape = shape;
                         layer.outputs[0].arguments[0].shape = shape;
                     });
+                    register(   22, 'RESIZE_NEAREST_NEIGHBOR', '', (layer, reader) => {
+                        layer.flags = reader.uint32();
+                        layer.inputs = [ reader.parameter('input') ];
+                        layer.outputs = [ reader.parameter('output') ];
+                        layer.inputs[0].arguments[0].shape = [ reader.uint32(), reader.uint32(), reader.uint32() ];
+                        layer.out_width = reader.uint32();
+                        layer.out_height = reader.uint32();
+                        layer.align_corners = reader.uint32();
+                    });
+                    register(   23, 'QUANTIZED_RESIZE_NEAREST_NEIGHBOR', '', (layer, reader) => {
+                        layer.flags = reader.uint32();
+                        layer.inputs = [ reader.parameter('input') ];
+                        layer.outputs = [ reader.parameter('output') ];
+                        layer.inputs[0].arguments[0].shape = [ reader.uint32(), reader.uint32(), reader.uint32() ];
+                        layer.out_width = reader.uint32();
+                        layer.out_height = reader.uint32();
+                        layer.align_corners = reader.uint32();
+                    });
                     register( 1000, 'CONV', 'Layer');
                     register( 1001, 'DWCONV', 'Layer');
                     register( 1002, 'QUANTIZED_RESHAPE', 'Shape');