4705ffe09b2442f34900b0b3185000478af44e92,src/megaface/gen_megaface.py,,get_feature,#Any#Any#Any#Any#Any#Any#Any#Any#,48

Before Change

      input_blob = np.expand_dims(_img, axis=0)
      net.arg_params["data"] = mx.nd.array(input_blob, net.ctx)
      net.arg_params["softmax_label"] = mx.nd.empty((1,), net.ctx)
      exe = net.sym.bind(net.ctx, net.arg_params ,args_grad=None, grad_req="null", aux_states=net.aux_params)
      exe.forward(is_train=False)
      _embedding = exe.outputs[0].asnumpy().flatten()
      //print(_embedding.shape)
      if embedding is None:

After Change

      //_img = nimg
      input_blob = np.expand_dims(_img, axis=0)
      data = mx.nd.array(input_blob)
      db = mx.io.DataBatch(data=(data,))
      net.model.forward(db, is_train=False)
      _embedding = net.model.get_outputs()[0].asnumpy().flatten()
      //print(_embedding.shape)

Instances