MobileFaecNet实现58个人脸关键点检测(附训练源码)

MobileFaecNet实现58个人脸关键点检测

1.0 caffe修改

实现新网络时,新设计的layer需要自己完成代码并添加到caffe中,同时由于人脸关键点是回归任务,因此为多标签数据,需要修改数据层image_data_layer从而可以支持多标签输入。(但是caffe之前都不支持多标签输入吗?)

image_data_layer.cpp最后几行原始代码:
MobileFaecNet实现58个人脸关键点检测(附训练源码)_第1张图片
image_data_layer.cpp最后几行修改后代码:
在这里插入图片描述
在caffe_path/include/caffe/layers路径下加入新增模块代码,其中需要注意的是,一般需要添加三个文件,分别是: .cpp .hpp .cu三个文件为后缀的文件,hpp和cpp文件的具体说明见此链接,cu文件是因为要用到cuda加速所配套的文件。(待补充)

参考链接

链接1
代码1
链接2
代码2

你可能感兴趣的:(MobileFaecNet实现58个人脸关键点检测(附训练源码))