MTCNN 复现

标签：cuda   das   failed   结果   style   erro   数据量   比较   hit   

0. 背景

　　最近有个猫脸检测任务，终极目标是实现猫脸和一些特征点的定位，从效率上来说，使用MTCNN比较合适，所以尝试训练了一下，并没有真正的复现。

1.踩过的坑

　　MTCNN这么火，所以从一开始就没打算自己写，直接在github上找的相关代码。平时习惯用 Pytorch， 没用过Tensorflow，所以就找了caffe的实现。

3.　结果

　　最后的结果上来说，对于正脸的猫是可以检测出来的，特征点也大约在定义的位置，由于数据量小，Onet 本身就不好训练，所以能跑出结果就暂时先放一放了。

4. 遇到的麻烦

　　主要是caffe的问题。

　　4.1 snapshot_perfix 参数不能传，只要加入这个参数就会运行失败，没解决只能不用这个参数了。

　　4.2 在训练 Lnet 的时候（修正Onet的landmarks）caffe 出错， math_functions.cu:79  Check failed: error==cudaSucess。没找到治本的解决办法，就先把caffe编译成cpu版本，训练了一下。

 5. 欢迎交流

MTCNN 复现

标签：cuda   das   failed   结果   style   erro   数据量   比较   hit   

原文地址：https://www.cnblogs.com/xiaochun-zhao/p/10283191.html