libfacedetection简单使用记录

1、源码下载

直接从github上克隆项目仓库。

git clone https://github.com/ShiqiYu/libfacedetection.git

2、编译

2.1、linux

这个项目使用了cmake脚本,先生成makefile。(我这里是在ArchLinux x86_64环境下测试的)

cmake -DENABLE_NEON=OFF -DCMAKE_BUILD_TYPE=RELEASE .

执行上面的命令成功后,执行下面语句进行编译

make -j4

编译完成后会同时生成动态库和静态库

[ 90%] Linking CXX static library libfacedetection.a
[100%] Linking CXX shared library libfacedetection.so

2.2、Windows MINGW64

这里和上面也是一样的。先使用cmake生产Makefile文件,然后执行编译即可。这里我指定了使用g++作为C++的编译器,因为如果不指定会使用gcc去链接,导致生成动态库的时候找不到c++标准库的一些定义。

cmake -DENABLE_NEON=OFF -DCMAKE_BUILD_TYPE=RELEASE  -DCMAKE_CXX_COMPILER=g++ .

其它的与上面linux的一致。

备注:我这里使用的是MSYS2环境,直接使用pacman命令安装的gcc/g++。

2.3、VS2017 NMake编译

因为就这么几个文件,生成一个庞大的VS工程实在有点多余,所以我简单写了一个Makefile.vc文件,使用nmake脚本来生成静态库(因为源码中并没有对函数接口进行export,所以只生成静态库)。

Makefile.vc 文件内容如下:

# 编译参数设置
CXX           = cl
DEFINES       = -DWIN32 -DWIN64 -DNDEBUG -D_WINDLL
CXXFLAGS      = -nologo -Zc:wchar_t -FS -Zc:rvalueCast -Zc:inline -Zc:strictStrings -Zc:throwingNew -Zc:referenceBinding -O2 -MD -W3 -w34100 -w34189 -w44996 -w44456 -w44457 -w44458 -wd4577 -wd4467 -EHsc $(DEFINES)
INCPATH       = -I.
LIBEXE        = lib
LIBFLAGS      = /NOLOGO 
LIBS          =  kernel32.lib

# 源文件
SOURCES       = facedetectcnn.cpp			\
				facedetectcnn-floatdata.cpp	\
				facedetectcnn-int8data.cpp	\
				facedetectcnn-model.cpp
OBJECTS       = facedetectcnn.obj			\
				facedetectcnn-floatdata.obj	\
				facedetectcnn-int8data.obj	\
				facedetectcnn-model.obj

# 输出目标文件
DESTDIR_TARGET = libfacedetection.lib

# 编译规则
.cpp.obj::
    $(CXX) @<< -c $(CXXFLAGS) $(INCPATH) $<
<<

all:	$(DESTDIR_TARGET)

$(DESTDIR_TARGET):	$(OBJECTS)
	$(LIBEXE) $(LIBFLAGS) /OUT:$(DESTDIR_TARGET) $(LIBS) $(OBJECTS)

facedetectcnn.obj: facedetectcnn.cpp facedetectcnn.h
facedetectcnn-floatdata.obj: facedetectcnn-floatdata.cpp facedetectcnn.h
facedetectcnn-int8data.obj: facedetectcnn-int8data.cpp facedetectcnn.h
facedetectcnn-model.obj: facedetectcnn-model.cpp facedetectcnn.h

clean:
	del /Q $(OBJECTS)

VS2017开发者命令行工具中直接使用nmake命令来生成静态库

nmake -f Makefile.vc

3、简单测试程序

使用Qt写了一个小程序,简单的测试了一下。

测试的时候发现年龄始终是0,然后就看了一下源码,发现其并未对年龄和面部特征点进行输出,实际在代码中也没有进行检测,年龄和面部特征点相关的代码我没有找到(2019年4月16日)。

3.1、测试截图

3.2、测试代码如下

/*****************************************************
 *       project:libfacedetection 测试代码            *
 *       anchor:ymwh@foxmail.com/solym@sohu.com      *
 *       date:2019-4-16                              *
 *****************************************************/

#include <facedetectcnn.h>

#include <QApplication>
#include <QWidget>
#include <QLabel>
#include <QLineEdit>
#include <QPushButton>
#include <QVBoxLayout>
#include <QHBoxLayout>
#include <QFileDialog>
#include <QPainter>
#include <QGraphicsView>

int main(int argc, char *argv[])
{
    QApplication a(argc, argv);
    QImage srcImage;
    // 创建窗口
    QWidget widget;
    // 添加控件
    QGraphicsScene* gsViewScene = new QGraphicsScene();
    gsViewScene->setItemIndexMethod(QGraphicsScene::NoIndex);
    QGraphicsView* gvImageView = new QGraphicsView(&widget);
    gvImageView->setScene(gsViewScene);

    QLineEdit* leImagePath = new QLineEdit(&widget);
    QPushButton* pbSelectFile = new QPushButton(QStringLiteral("选择文件"),&widget);
    QPushButton* pbRunDetect = new QPushButton(QStringLiteral("执行检测"),&widget);
    pbRunDetect->setEnabled(false);
    QHBoxLayout* hbLayout = new QHBoxLayout;
    // 设置布局
    hbLayout->addWidget(leImagePath);
    hbLayout->addWidget(pbSelectFile);
    hbLayout->addWidget(pbRunDetect);
    QVBoxLayout* vbLayout = new QVBoxLayout(&widget);
    vbLayout->addLayout(hbLayout);
    vbLayout->addWidget(gvImageView);
    // 添加处理操作
    QObject::connect(pbSelectFile,&QPushButton::clicked,
                     [leImagePath,gvImageView,gsViewScene,pbRunDetect,&srcImage,&widget]()
    {
        static QString path;
        path = QFileDialog::getOpenFileName(&widget,
                                            QStringLiteral("选择人像图"),
                                            path,
                                            QString("Images (*.png *.jpg *.jpeg *.jfif)"));
        if(path.isEmpty()){return;}
        QImage image;
        if(!image.load(path)){return;}
        // 图像太大的时候,执行太慢,先缩小一点
        if(image.width() > 2048){ image = image.scaledToWidth(2048); }
        if(image.width() > 1536){ image = image.scaledToWidth(1536); }

        srcImage.swap(image);
        leImagePath->setText(path);

        gsViewScene->clear();
        gsViewScene->setSceneRect(srcImage.rect());
        gsViewScene->addPixmap(QPixmap::fromImage(srcImage));
        gvImageView->fitInView(gvImageView->sceneRect());

        pbRunDetect->setEnabled(true);
    });

    QObject::connect(pbRunDetect,&QPushButton::clicked,
                     [gvImageView,gsViewScene,&srcImage]
    {
        // 转换为RGB24
        QImage image = srcImage.convertToFormat(QImage::Format_RGB888);
        int rows = image.height();
        int cols = image.width();
        int rowbytes = image.bytesPerLine();

        uchar* pImgData = image.bits();
        // RGB -> BGR 因为facedetect_cnn函数要求传入图像为BGR三波段图像
        for( int r = 0; r < rows; ++r ){
            uchar* pRow = pImgData + (r * rowbytes);
            for(int c = 0; c < cols; ++c ){
                qSwap(pRow[c*3],pRow[c*3+2]);
            }
        }
        // 进行检测
        QByteArray buffer(0x20000,0);
        int* pResults = facedetect_cnn((uchar*)buffer.data(),pImgData,cols,rows,rowbytes);
        // 将检测结果画到图像上
        QPixmap pixmap = QPixmap::fromImage(srcImage);
        if(pResults != NULL) {
            for(int i=0; i< *pResults; ++i) {
                short * p = ((short*)(pResults+1))+142*i;
                int x = p[0];
                int y = p[1];
                int w = p[2];
                int h = p[3];
                int confidence = p[4];
                // 查看源码可知,其并未进行人像框范围和置信率以外的数据赋值
                int angle = p[5];
                QPainter painter(&pixmap);
                // painter.setBrush(QBrush(QColor(255,0,0),));
                painter.setPen(Qt::green);
                // painter.drawEllipse(x,y,w,h);
                painter.drawRect(x,y,w,h);
                painter.drawText(x,y,w,h,Qt::AlignCenter,
                                 QStringLiteral("置信:%1%\n年龄:%2").arg(confidence).arg(angle));
            }
        }

        gsViewScene->clear();
        gsViewScene->setSceneRect(pixmap.rect());
        gsViewScene->addPixmap(pixmap);
        gvImageView->fitInView(gvImageView->sceneRect());
    });

    widget.resize(1024,768);
    widget.show();
    return a.exec();
}
posted @ 2019-04-16 12:28  乌合之众  阅读(4290)  评论(3编辑  收藏  举报
clear