[Programming / OpenCV] 1초마다 지정 폴더에 이미지 저장하기

다음 코드는 웹캠으로부터 입력받은 Frame Capture 이미지의 좌측 상단에 현재의 현재의 날짜와 시간을 출력하고, 이미지를 해당 경로(D:/WebCam/)에 1초 간격으로 이미지를 저장하는 OpenCV 코드입니다. 이미지를 일정한 시간 간격으로 저장하는데 유용한 코드입니다.

WebCamp.cpp

/**
 * @file WebCam_Demo.cpp
 * @brief Sample code showing how to save images in 1 sec. interval
 * @author Geol Choi, ph.D
 */
 
#include <opencv2/opencv.hpp>
#include "opencv2/imgproc/imgproc.hpp"
#include "opencv2/highgui/highgui.hpp"
#include <stdlib.h>
#include <stdio.h>
#include <iostream>
#include <windows.h>
#include <ctime>
 
using namespace std;
using namespace cv;
 
void CreateFolder( const char* path );
 
/* @ function main */
int main( int argc, char** argv )
{
    // open the default camera
    int device = 0;
    cv::VideoCapture cap( device );
 
    std::string strFilePath = "";
    std::clock_t start;
    double duration;
    
    int nNew_duration = 0;
    int nOld_duration = 0;
 
    // make directory
    std::time_t rawtime;
    std::tm* timeinfo;
    char c_buffer[80];
    char c_pathBuffer[80];
 
    std::time( &rawtime );
    timeinfo = std::localtime( &rawtime );
 
    std::strftime( c_buffer, 80, "%Y%m%d", timeinfo );
 
    std::string strBasePath( c_buffer );
    sprintf_s( c_pathBuffer, "D:\\WebCam\\%s", strBasePath );
 
    CreateFolder( c_pathBuffer );
 
    std::string basePath( c_pathBuffer );
 
    start = std::clock();
 
    // check if we succeeded
    if( !cap.isOpened() )
        return -1;
 
    cv::namedWindow( "WebCam Frame Capture", 1 );
 
    /// Text Location
    cv::Point myPoint;
    myPoint.x = 10;
    myPoint.y = 40;
 
    /// Font Face
    int myFontFace = 2;
 
    /// Font Scale
    double myFontScale = 1.2;
 
    int tmp = 1;
    bool bTmp = false;
 
    for( ; ; ) {
        if( !bTmp ) {
            std::cout << "Please enter any key to start video capture: " << std::endl;
            std::cin >> tmp;
            bTmp = true;
        }
 
        cv::Mat frame;
 
        // current date/time based on current system
        std::time( &rawtime );
 
        timeinfo = std::localtime( &rawtime );
        std::strftime( c_buffer, 80, "%Y-%m-%d-%H:%M:%S", timeinfo );
    
        // convert now to string form
        std::string myText( c_buffer );
 
        // another time format
        std::strftime( c_buffer, 80, "%H%M%S", timeinfo );
        std::string str( c_buffer );
        
        // get a new frame from camera
        cap >> frame;
 
        cv::putText( frame, myText, myPoint, myFontFace, myFontScale, cv::Scalar( 255, 128, 0 ) );
 
        cv::imshow( "WebCam Frame Capture", frame );
 
        duration = ( std::clock() - start ) / (double) CLOCKS_PER_SEC;
        nNew_duration = int( duration );
 
        if( nNew_duration - nOld_duration > 0 ) {
            std::stringstream ss;
            ss << basePath << "\\" << str << ".png";
            std::string s_ = ss.str();
 
            cv::imwrite( s_, frame );
        }
 
        nOld_duration = nNew_duration;
 
        if( cv::waitKey( 10 ) >= 10 ) break;
    }
 
    // the camera will be deinitialized automatically in VideoCapture destructor
    return 0;
}
 
 
/* @ function CreateFolder */
void CreateFolder( const char* path )
{
    wchar_t wtext[20];
    std::mbstowcs( wtext, path, strlen( path ) + 1 );//Plus null
    LPWSTR ptr = wtext;
 
    if( !CreateDirectory( ptr, NULL ) )
    {
        return;
    }
}

저장된 이미지 출력