yamada
1
I use mxnet on insightface to extract features.
The output of the model is a feature, but insightface sets the label for forward().
db = mx.io.DataBatch(data=(_data,), label=(_label,))
model.forward(db, is_train=False)
and did not enter a label when loading the model.
model.bind(data_shapes=[(‘data’, (args.batch_size, 3, image_size[0], image_size[1]))])
I don’t think you should enter a label in the feature extraction to avoid errors.