retinaface icon indicating copy to clipboard operation
retinaface copied to clipboard

请问源项目中的R50模型要如何转换成caffe模型?

Open fhjjack opened this issue 5 years ago • 16 comments

fhjjack avatar Aug 09 '19 08:08 fhjjack

R50模型转换可以看下有没有不支持的层,如果有的话可能不能直接转。需要自己写模型转换工具。

clancylian avatar Aug 14 '19 03:08 clancylian

@clancylian 我转出来,在caffe上跑,结果不一样,面部框变大了,而且5个关键点也不准确。你转出来的结果和官方一样嘛?

xiakj avatar Oct 30 '19 05:10 xiakj

@xiakj @clancylian 我这边实验也是,代码没改,mnet0.25跑出的结果没有问题;但是resnet50转过来的caffemodel,检测不到人脸。这是什么情况呢??

OPPOA113 avatar Nov 07 '19 04:11 OPPOA113

resnet50的转换方法参考:https://github.com/szad670401/advanced-mxnet2caffe

OPPOA113 avatar Nov 07 '19 04:11 OPPOA113

@OPPOA113 现在检测结果正确了,一定要保证测试图像比例不要发生变化,可以通过调整*_deploy.prototxt中的data层的dims参数解决,也可以在源码中设置。

xiakj avatar Nov 08 '19 01:11 xiakj

@xiakj 谢谢您。 请问一下, 1、你用的是哪个工程转的caffemodel? 2、这个图像比例设置的是多少,参考的是哪里的?

OPPOA113 avatar Nov 08 '19 01:11 OPPOA113

@OPPOA113

  1. 就是上面你提到的那个advanced-mxnet2caffe。也有已经转换号的参考https://github.com/Charrin/RetinaFace-Cpp
  2. 我输入图像是1280720的,因为我的gpu性能比较低,所以我按照比例将dims设置为360640了

xiakj avatar Nov 08 '19 01:11 xiakj

@xiakj @clancylian 采用这个工程下的,通样的代码,图像都是resize到1280x720的输入,mnet0.25检测的结果正常,RestNet50检测的结果不正常,这可能是什么地方出问题了? image

OPPOA113 avatar Nov 08 '19 02:11 OPPOA113

@OPPOA113 不太了解,感觉RestNet50模型本身的问题吧,你再好好读读RestNet50的prototxt

xiakj avatar Nov 08 '19 02:11 xiakj

@xiakj 你用的R50的caffemodel和prototxt能不能发过来给我。我试试。[email protected]

OPPOA113 avatar Nov 08 '19 02:11 OPPOA113

@OPPOA113 请查收

xiakj avatar Nov 08 '19 02:11 xiakj

@xiakj 谢谢,已收到。 但是检测的结果还是不对。 mnet0.25 模型检测是没有问题的,不知道resnet50检测结果为何不正确.......

OPPOA113 avatar Nov 08 '19 02:11 OPPOA113

@xiakj @OPPOA113 我使用https://github.com/Charrin/RetinaFace-Cpp下方法转换后的R50,也是无法检测出来。请问二位解决这个问题了吗?能否发送我一份模型和prototxt,谢谢[email protected]

nuanxinqing avatar Mar 04 '20 12:03 nuanxinqing

@nuanxinqing 您可以试一下github.com/szad670401/advanced-mxnet2caffe,我是用这个转换出来的。但是caffemodel转换过程中也遇到了一些问题,您留意一下,也可以再沟通。

xiakj avatar Mar 06 '20 01:03 xiakj

这里 https://github.com/cholihao/Retinaface-caffe 找到了一个google drive链接, 里面有caffe的模型: https://drive.google.com/drive/folders/1VoABSiHXiVlRCEryKtf3UG_BW9236UUp

shangdibufashi avatar Mar 30 '20 15:03 shangdibufashi

@jiapinai 已经好了,谢谢您!

xiakj avatar Mar 31 '20 03:03 xiakj