insightface 使用recognition/tools/mask_renderer.py 中加口罩方法出错, 找不到retinaface_r50

使用recognition/tools/mask_renderer.py 中加口罩方法出错, 找不到retinaface_r50_v1.onnx模型文件

Open gangeqian opened this issue 3 years ago • 10 comments

使用recognition/tools/mask_renderer.py 中加口罩方法出错

就是在~/.insightface/models/retinaface_r50_v1/ 目录下找不到retinaface_r50_v1.onnx模型文件。官方只提供了mxnet的模型文件

Jun 18 '21 07:06 gangeqian

我基本确定到问题了，屏蔽掉人脸检测，只加口罩是没问题的， insightface/recognition/tools/mask_renderer.py 中是先检测人脸再加口罩。问题出在人脸检测上，代码会从~/.insightface/models/下寻找retinaface_r50_v1的onnx模型。而官方只提供了retinaface_r50_v1的mxnet模型，insightface 库 <0.1.5支持MXNET模型，而新的0.3.0 版本换成了onnx支持。

Jun 18 '21 07:06 gangeqian

我有尝试将retinaface_r50_v1 的mxnet 模型转换成onnx模型，但是会报softmax哪个算子不识别出错。

Jun 18 '21 07:06 gangeqian

please use the latest code

Jun 19 '21 01:06 nttstar

我使用新版insightface(0.3.5)進行測試，模型部分下載過 "antelope" 與 ""antelopev2"，搭配底下BFM.m/BFM_UV.mat檔案 BFM.m: https://drive.google.com/file/d/1In3O2FF9CwuzNBUS83u441qo7OFr0273/view?usp=sharing BFM_UV.mat: https://drive.google.com/file/d/1SL0HanE4yUkU94qEG0vQBDPbGBxarLmO/view?usp=sharing

測試程式碼如下:

tool = MaskRenderer()
tool.prepare(ctx_id=0, det_size=(128,128))

image = cv2.imread("./Tom_Hanks_54745.png")
mask_image  = "mask_blue"
params = tool.build_params(image)
mask_out = tool.render_mask(image, mask_image, params)

cv2.imwrite('output_mask.jpg', mask_out)

但結果如下圖並不理想: output_mask