OpenCV实现抠图工具

#include <io.h>
#include <conio.h>
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <opencv.hpp>
 
// 抠图是单目标还是多目标，若为单目标请将下面这行文字取消注释，反之请注释这段文字。
// #define SINGLE_OBJECT
 
#define TRUE 1     // 逻辑真
#define FALSE 0   // 逻辑假
 
#define CODE_ESC 27     // ESC 键的编码
#define CODE_SPC 32    // 空格键的编码
 
#define STATUS_WAIT 0  // 抠图等待状态
#define STATUS_PROC 1     // 抠图进行状态
#define STATUS_DONE 2    // 抠图完成状态
 
#define VIDEO_FILENAME "capture-1.mp4"          // 视频流文件名
 
static int m_x1     = 0;    // 鼠标指针坐标（起点 x）
static int m_x2     = 0;    // 鼠标指针坐标（终点 x）
static int m_y1     = 0;    // 鼠标指针坐标（起点 y）
static int m_y2     = 0;     // 鼠标指针坐标（终点 y）
static int m_status = STATUS_WAIT;              // 当前抠图状态指示
 
static void on_mouse(int, int, int, int, void*);// 鼠标回调
 
// 主程序
int main(void)
{
    int        end    = 0;         // 指示是否结束程序
    int        next   = 0;         // 指示是否切换到下一张图片
    int        code   = 0;        // 存储按键编码
    int        count  = 0;        // 存储目标计数
    int        frame  = 0;         // 视频帧号（用于间隔采样）
    int        maxCol = 0;             // 图像最大列数（= 图像宽度 - 1）
    int        maxRow = 0;                      // 图像最大行数（= 图像高度 - 1）
    CvCapture* pVideo = NULL;          // 视频流对象
    IplImage*  pFrame = NULL;           // 视频帧图像（用于样本存储）
    IplImage*  pFrmCp = NULL;           // 视频帧图像（用于屏幕显示）
    CvPoint    pt1    = cvPoint(0, 0);          // 矩形框对角坐标点 1
    CvPoint    pt2    = cvPoint(0, 0);          // 矩形框对角坐标点 2
    CvRect     r      = cvRect(0, 0, 0, 0);     // 感兴趣区域矩形框
    char       seq[]  = "-2147483648";          // 目标计数的字串形式
    char       fil[]  = "data\\-2147483648.jpg";// 文件名字串
 
    // 载入视频流
    pVideo = cvCreateFileCapture(VIDEO_FILENAME);
    if (!pVideo)
    {
        return -1;
    } // if (!pVideo)
 
    // 创建数据存储目录
    if (_access("data", 0) != 0)
    {
        system("md data");
    } // if (_access())
 
    // 获取首帧图像，并创建拷贝，同时得到最大列数和行数，方便之后使用
    pFrame = cvQueryFrame(pVideo);
    if (pFrame)
    {
        pFrmCp = cvCreateImage(cvGetSize(pFrame), 8, pFrame->nChannels);
        maxCol = pFrmCp->width - 1;
        maxRow = pFrmCp->height - 1;
    } // if (pFrame)
    else
    {
        cvReleaseCapture(&pVideo);
        return -1;
    } // else
 
    // 设置显示窗口，并设置鼠标回调
    cvNamedWindow("Monitor", CV_WINDOW_AUTOSIZE);
    cvSetMouseCallback("Monitor", on_mouse, NULL);
 
    // 其他初始化
    end = FALSE;
    count = 0;
    frame = 0;
 
    while (!end && pFrame)
    {
        next = FALSE;
        while (!next && !end)
        {
            // 将原始视频图像复制到拷贝区域中（清除已将图像进行污染的线条、矩形框等）
            cvCopy(pFrame, pFrmCp, NULL);
            if (STATUS_WAIT == m_status)
            {
                // 等待抠图状态。画出横向和纵向的参考线
                cvLine(pFrmCp, cvPoint(m_x1, 0), cvPoint(m_x1, maxRow), CV_RGB(0, 255, 0));
                cvLine(pFrmCp, cvPoint(0, m_y1), cvPoint(maxCol, m_y1), CV_RGB(0, 255, 0));
            } // if (STATUS_WAIT)
            else if (STATUS_PROC == m_status)
            {
                // 抠图过程中。画出当前选定的感兴趣区域
                pt1 = cvPoint(m_x1, m_y1);
                pt2 = cvPoint(m_x2, m_y2);
                cvRectangle(pFrmCp, pt1, pt2, CV_RGB(0, 255, 0));
            } // else if (STATUS_PROC)
            else if (STATUS_DONE == m_status)
            {
                // 抠图完毕，获得感兴趣区域并按编号保存样本
                r = cvRect(
                    m_x1,
                    m_y1,
                    m_x2 - m_x1 + 1,
                    m_y2 - m_y1 + 1
                    ); // 矩形感兴趣区域
                if (r.width > 30 && r.height > 30)
                {
                    // 区域达到了一定大小，抠图有效，保存感兴趣区域样本
                    ++count;
                    cvSetImageROI   (pFrame, r);
                    sprintf_s       (seq, "%d", count);
                    strcpy_s        (fil, "data\\");
                    strcat_s        (fil, seq);
                    strcat_s        (fil, ".jpg");
                    cvSaveImage     (fil, pFrame, 0);
                    cvResetImageROI (pFrame);
 
#ifdef SINGLE_OBJECT
                    m_next = TRUE;
#endif
                } // if (r.width)
                
                // 恢复抠图等待状态
                m_status = STATUS_WAIT;
            } // else if (STATUS_DONE)
 
            cvShowImage("Monitor", pFrmCp);
            code = cvWaitKey(10);
            if (CODE_SPC == code)
            {
                next = TRUE;
            } // if (CODE_SPC)
            else if (CODE_ESC == code)
            {
                end = TRUE;
            } // else if (CODE_ESC)
        } // while (!next)
 
        if (next)
        {
            do
            {
                pFrame = cvQueryFrame(pVideo);
                ++frame;
            } while (pFrame && frame % 60 != 0); // do...while
        } // if (next)
    } // while (!end)
 
    cvDestroyAllWindows();
    cvReleaseImage(&pFrmCp);
    cvReleaseCapture(&pVideo);
 
    return 0;
} // main()
 
 
// 鼠标事件回调
void on_mouse(int event, int x, int y, int flags, void* param)
{
    switch (flags)
    {
    case CV_EVENT_MOUSEMOVE:
        if (STATUS_WAIT == m_status)
        {
            // 等待状态，确定感兴趣区域起点
            m_x1 = x, m_y1 = y;
        } // if (STATUS_WAIT)
        else if (STATUS_PROC == m_status)
        {
            // 捕捉状态，确定感兴趣区域终点
            m_x2 = x, m_y2 = y;
        } // else if (STATUS_PROC)
        break;
 
    case CV_EVENT_LBUTTONDOWN:
        if (STATUS_WAIT == m_status)
        {
            // 等待状态按下鼠标，进入捕捉状态，固定起点
            m_x1 = x, m_y1 = y;
            m_status = STATUS_PROC;
        } // if (STATUS_WAIT)
        else if (STATUS_PROC == m_status)
        {
            // 捕捉状态按下鼠标，捕捉完成，固定终点
            m_x2 = x, m_y2 = y;
            m_status = STATUS_DONE;
        } // else if (STATUS_PROC)
        break;
    } // switch
} // on_mouse()