insightface icon indicating copy to clipboard operation
insightface copied to clipboard

miss import in Insightface python package

Open woreom opened this issue 4 years ago • 1 comments
trafficstars

At model_zoo.face_detection code imports mxnet and mxnet.ndarray inside FaceDetector init function which makes the code unusable for example,

from insightface.model_zoo.face_detection import retinaface_r50_v1
model = retinaface_r50_v1()
model.prepare(ctx_id = -1, nms=0.4)

which will result in

NameError: name 'mx' is not defined

importing them at top of the module will fix the issue or doing any correct way of local importing since model_zoo.get_model has changed and can't be used to use the models anymore, this was the easiest way I could find to load models

woreom avatar Jun 14 '21 03:06 woreom