MTCNN试用

检测工作想借用MTCNN里的48-net,源码来自git 下下来就能跑,真是良心

进入pepare_data准备好数据以后进入48-net,目录下有一个pythonLayer.py,由于loss函数是自定义的python层,所以train.sh里,第一行要把这个目录export到pythonpath

然后在train.sh第二行里配上自己的caffe编译路径,运行,报错找不到python层,查到是因为给的caffe路径不支持python层导致的,这是在caffe编译的时候没有选择 WITH_PYTHON_LAYER := 1

caffe路径选为py-faster-rcnn/caffe-fast-rcnn/build/tools/caffe 就可以了。

猜你喜欢

转载自www.cnblogs.com/zhengmeisong/p/9265238.html
今日推荐