OpenCV與Socket實現樹莓派獲取攝像頭視頻至電腦

OpenCV能夠爲我們帶來便捷的圖像處理接口，但是其處理速度在一塊樹莓派上肯定是不盡如人意的。尤其當我們想要使用複雜的算法時，只能把算法託到服務器上纔有可能。這裏介紹了一種方法，實現樹莓派傳輸Mat至電腦。

準備工作

1、配置好樹莓派上的OpenCV.
2、配置好電腦上的OpenCV.

思路

Socket實現方法較爲簡單，但在此處需要注意的是：
樹莓派上的，是linux系統；而我電腦上是windows系統，需要注意這一點來進行編程。

代碼

樹莓派作爲客戶端，發送Mat.

#include <stdio.h>
#include <stdlib.h>
#include <unistd.h>
#include <string.h>
#include <sys/types.h>
#include <sys/socket.h>
#include <netinet/in.h>
#include <netdb.h> 
#include <opencv2/opencv.hpp>
using namespace cv;
using namespace std;

void error(const char *msg)
{
    perror(msg);
    exit(0);
}

int main()
{
    int sockfd, portno, n;
    struct sockaddr_in serv_addr;
    struct hostent *server;

    portno = atoi("8888");
    sockfd = socket(AF_INET, SOCK_STREAM, 0);
    if (sockfd < 0)
        error("ERROR opening socket");
    server = gethostbyname("10.138.216.104");//這裏填IP地址
    if (server == NULL) {
        fprintf(stderr, "ERROR, no such host\n");
        exit(0);
    }
    bzero((char *)&serv_addr, sizeof(serv_addr));
    serv_addr.sin_family = AF_INET;
    bcopy((char *)server->h_addr,
        (char *)&serv_addr.sin_addr.s_addr,
        server->h_length);
    serv_addr.sin_port = htons(portno);
    if (connect(sockfd, (struct sockaddr *) &serv_addr, sizeof(serv_addr)) < 0)
        error("ERROR connecting");

    VideoCapture cap(0);
    Mat frame;
    while (true)
    {
        cap >> frame;
        cvtColor(frame, frame, CV_BGR2GRAY);
        // Send data here
        int bytes = send(sockfd, frame.data, frame.total()*frame.elemSize(), 0));
    }
    waitKey(0);
    return 0;
}

服務器接收並顯示

    #include <winsock2.h>  
    #include <opencv.hpp>
    using namespace cv;
    using namespace std;
    #pragma comment(lib,"ws2_32.lib")  
    #define imgSize 307200 //這裏需要算一下，因爲Visual Studio不能支持用變量初始化數組
    //imgSize=frame.total()*frame.elemSize();
    //即：如果爲灰度圖：圖像的寬*圖像的高 這裏是640*480

    int main(int argc, char* argv[])
    {
    //初始化WSA  
    WORD sockVersion = MAKEWORD(2, 2);
    WSADATA wsaData;
    if (WSAStartup(sockVersion, &wsaData) != 0)
    {
        return 0;
    }
    //創建套接字  
    SOCKET slisten = socket(AF_INET, SOCK_STREAM, IPPROTO_TCP);
    if (slisten == INVALID_SOCKET)
    {
        printf("socket error !");
        return 0;
    }
    //綁定IP和端口  
    sockaddr_in sin;
    sin.sin_family = AF_INET;
    sin.sin_port = htons(8888);
    sin.sin_addr.S_un.S_addr = INADDR_ANY;
    if (bind(slisten, (LPSOCKADDR)&sin, sizeof(sin)) == SOCKET_ERROR)
    {
        printf("bind error !");
    }

    //開始監聽  
    if (listen(slisten, 5) == SOCKET_ERROR)
    {
        printf("listen error !");
        return 0;
    }

    SOCKET sClient;
    sockaddr_in remoteAddr;
    int nAddrlen = sizeof(remoteAddr);

    cout << "Wait.." << endl;
    do
    {
        sClient = accept(slisten, (SOCKADDR *)&remoteAddr, &nAddrlen);
    } while (sClient == INVALID_SOCKET);
    cout<<"Received Information come from"<< inet_ntoa(remoteAddr.sin_addr)<<endl;

    int bytes = 0;
    while (1)
    {
        char socketData[imgSize];

        for (int i = 0; i < imgSize; i += bytes) {
            if ((bytes = recv(sClient, socketData + i, imgSize - i, 0)) == -1)
            {
                cout << "!Fault" << endl;
                exit(-1);
            }
        }

        // change the last loop to below statement
        Mat img(Size(640, 480), CV_8UC1, socketData);//根據攝像頭大小自行修改
        imshow("Face_Socket", img);
        waitKey(1);
    }
    closesocket(slisten);
    WSACleanup();
    return 0;
}

先運行服務器，後運行客戶端（其實無所謂），我們就可以在服務器上看到樹莓派通過攝像頭獲取到的幀，那麼就可以對齊進行很多操作了。