腾讯NCNN运行Mobilenet-SSD检测视频

腾讯NCNN运行Mobilenet-SSD检测视频

1 源码下载与修改

git clone https://github.com/GuoJaw/MobileNetSSD_NCNN

下载后的文件列表如下
将自己的 mobilenet.param 和 mobilenet.bin 放在 kitti_model 目录下

1. 修改CmakeLists.txt

cmake_minimum_required(VERSION 2.8) # cmake verson，指定cmake版本 
add_definitions(-std=c++11 -g)

project (ncnn_test)


set(ncnn_test_INCLUDE_DIRS /home/chris/workplace/MobileNetSSD_NCNN/include)  # 文件路径

include_directories(
	${ncnn_test_INCLUDE_DIRS} 

)

### OpenCV的配置 ###
set (OpenCV_DIR /usr)  
find_package(OpenCV 3 REQUIRED)   # OpenCV版本 本文安装的是OpenCV3.3 

add_executable (ncnn_test main.cpp)


set (NCNN_LIBS /home/chris/workplace/MobileNetSSD_NCNN/libncnn.a)  # 文件路径
set (GOMP_LIB /usr/lib/x86_64-linux-gnu/libgomp.so.1)

target_link_libraries( ncnn_test 
	${OpenCV_LIBS} 
	${NCNN_LIBS} 
	${GOMP_LIB}
)

2. 修改main.cpp

const char* class_names[] = { "background","your name"};

...

    cv::VideoCapture cap("/home/chris/workplace/MobileNetSSD_NCNN/video/test.mp4");

    if (!cap.isOpened())
    {
        std::cout << "video is not open" << std::endl;
        return -1;
    }

    cv::Mat frame;
    ncnn::Net mobilenet;

    mobilenet.load_param("/home/chris/workplace/MobileNetSSD_NCNN/kitti_model/mobilenet.param");
    mobilenet.load_model("/home/chris/workplace/MobileNetSSD_NCNN/kitti_model/mobilenet.bin");

2 编译运行

依次运行以下指令即可

mkdir build
cd build
cmake ..
make
./ncnn_test