Opencv将目录下的图片存储为视频

源码

/*************************************************************************
    > File Name: main.cpp
    > Author: 
    > Mail: 1@163.com 
    > Created Time: 2022年06月17日 星期五 14时00分37秒
    > g++ main.cpp -o main `pkg-config --cflags --libs opencv`
 ************************************************************************/
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <dirent.h>
#include <unistd.h>
#include <iostream>
#include <vector>
#include <libgen.h>

#include <opencv2/opencv.hpp>  
#include <iostream>
#include <unistd.h>

using namespace cv;  
using namespace std;

vector <string> relativePath;
vector <string> fileName;


/******************************************************************************
 *
 * 功能：
 *		获得 (文件名.后缀) 和 (文件名) 和 (文件后缀)，Windows和Linux通用！
 *
 * 参数：
 *		_filePaht：			C语言字符指针，文件路径字符串
 *		_fullFileName：		C++字符串引用，获得[(文件名.后缀) 或 (文件名)]返回
 *		_fileName：			C++字符串引用，获得(文件名.后缀)返回
 *		_fileSuffix：		C++字符串引用，获得(后缀)返回
 *		_dot：				true，后缀带点(.txt)；false，后缀不带点(txt)；默认值为true
 *
 * 返回值：
 *		有后缀返回"."的索引；没有后缀则返回-1
 *
 ******************************************************************************/
int get_fileName_fileSuffix(const char *_filePaht, std::string &_fullFileName, std::string &_fileName, std::string &_fileSuffix, bool _dot = true) {

	// 合法性检查
	if (!_filePaht) return -1;
	std::string str = _filePaht;
	if (str.empty()) return -1;

	// .txt    的情况
	if (str.at(0) == '.') return -1;
	// file.   的情况
	if (str.at(str.size() - 1) == '.') return -1;



	char c = '\0';

	// 区分此函数是在Windows环境调用还是Linux环境调用
#if defined (_WIN64) || defined (WIN32) || defined (_WIN32)
	//printf("---Windows---\n");
	c = '\\';
#else
	//printf("---Linux---\n");
	c = '\/';
#endif

	// 去除字符串中的路径，剩下文件名
	std::string filename = strrchr(_filePaht, c) == NULL ? "" : strrchr(_filePaht, c) + 1;
	if (filename == "") {
		/* 说明字符串是file.txt或者file形式 */
		filename = _filePaht;

	} else {
		/* 说名字符串是C:\\abc\\file.txt或者C:\\abc\\file形式 */
	}

	// 找到 . 的位置
	size_t _size = filename.rfind(".");
	// 获得文件名，不包含后缀
	_fileName = filename.substr(0, _size);

	// 获得文件后缀
	std::string strsuffix = "";
	if (_size != -1) {	// 不等于-1说明有后缀 file.txt形式

		// true后缀带点，false后缀不带点
		if (_dot == true) {
			strsuffix = filename.substr(_size);     // 文件后缀 .txt

		} else {
			strsuffix = filename.substr(_size + 1);     // 文件后缀 txt
		}
	}
	// 后缀
	_fileSuffix = strsuffix;

	// 获得文件名.后缀(file.txt | file)
	_fullFileName = filename;

	return _size;
}

int readFileList(char *basePath)
{
    DIR *dir;
    struct dirent *ptr;
    char base[1000];

    if ((dir=opendir(basePath)) == NULL)
    {
        perror("Open dir error...");
        exit(1);
    }

    while ((ptr=readdir(dir)) != NULL)
    {
        if(strcmp(ptr->d_name,".")==0 || strcmp(ptr->d_name,"..")==0)    ///current dir OR parrent dir
            continue;
        else if(ptr->d_type == 8){    ///file
            printf("d_name:%s/%s\n",basePath,ptr->d_name);
            // 文件名
            string s1 = ptr->d_name;
            fileName.push_back(s1);
            // 全路径
            string s = basePath;
            s = s + "/";
            s = s + +ptr->d_name;
            relativePath.push_back(s);
            
            std::string fullFilename, filename, fileSuffix;
	          int r = get_fileName_fileSuffix(s1.c_str(), fullFilename, filename, fileSuffix, true);
	         
        }else if(ptr->d_type == 10){    ///link file
            printf("d_name:%s/%s\n",basePath,ptr->d_name);
        }else if(ptr->d_type == 4)    ///dir
        {
            memset(base,'\0',sizeof(base));
            strcpy(base,basePath);
            strcat(base,"/");
            strcat(base,ptr->d_name);
            readFileList(base);
        }
    }
    closedir(dir);
    return 1;
}

int main(int argc, const char *argv[])
{
    DIR *dir;
    #if 0
    char basePath[1000];
    memset(basePath,'\0',sizeof(basePath));
    // 得到当前的绝对路径
    getcwd(basePath, 999);
    #else
    char *basePath = (char *)argv[1];
    #endif
    
    // 获取目录下的全部文件
    //readFileList(basePath);
    
    
  VideoWriter w1;
  int encode_type = VideoWriter::fourcc('M', 'J', 'P', 'G'); 
 	bool isColor = true; 
	double fps = 25.0;
	string filename = "test_save.avi";
	w1.open(filename, encode_type, fps, cv::Size(1280, 720), isColor);  
 
	if (!w1.isOpened()) {     
		cout << "视频打开失败" << endl;
	}


    int id = 0;
    int maxNum = 300;
    for (int i = 0; i<maxNum; i++)
    {
      string filePathName = argv[1] ;
      filePathName = filePathName+ "/" + to_string(i) + ".png";

      if (! access(filePathName.c_str() ,0) )
      {
          Mat img;
          img = imread(filePathName.c_str());  
          if (img.empty())  //判断图像文件是否存在
          {
            cout << "读取失败" << endl;
            if (w1.isOpened())
              w1.release();
            return -1;
          }
          w1.write(img);
          imshow("test", img);  //显示图像
          waitKey(0); 
      }
    
    }
    return 0;
}

编译

 g++ img2video.cpp -o img2video `pkg-config --cflags --libs opencv`

运行

./img2video  ../results