需求分析:
公司需要做鉴黄项目,需要大量的图片数据资源,所以选择从视频上面截取,遍历电脑某个文件夹下面的所有文件夹里面的所有视频,然后做帧切割,假如算出全部视频的总帧数是100万,而我需要50万张图,就是每2帧切一张,也就是 100万/50万 = 2
下面是具体实现:
遍历文件夹(可以有多级文件夹)获取视频列表方法:
vector<String> getFilePath() {
String dir_path = "F:\qietu\\video";
vector<String> videos_list; // opencv 2 String 改为 string即可
/*
函数名:glob
参数:String pattern 字符串,由文件夹路径和所要读取的文件名的格式构成的一个正则模板
参数: std::vector<String>& result 字符串容器,存放所有符合模板的路径
参数:bool recursive = false 标志位,是否使用递归的方式(没发现使用true和false的结果有什么区别)
功能描述:遍历指定的文件夹,读取符合搜索模板的所有文件路径
*/
glob(dir_path, videos_list,true);
for (size_t j = 0; j < videos_list.size(); ++j)
{
string fileName = videos_list[j];
}
return videos_list;
}
获取上面全部视频列表加起来的总帧数
vector<String> listPath = getFilePath();
long allTotalFrame = 0;
for (int i = 0; i < listPath.size(); i++) {
VideoCapture cap;
cap.open(listPath[i]);
// 获取整视频帧数
long totalFrame = cap.get(CV_CAP_PROP_FRAME_COUNT);
allTotalFrame = allTotalFrame + totalFrame;
cout << "第 "<<i<<" 个视频宗贞数=" << totalFrame << endl;
}
cout << "总视频宗贞数=" << allTotalFrame << endl;
图片切割实现
for (;;)
{
Mat frame;
cap >> frame;
if (frame.empty())
{
break;
}
char* imageSaveName = new char[200];
//将图片的路径及名称输入到imageSaveName中
sprintf(imageSaveName, "F:\\test\\%05d.jpg", imgIndex);
cout << "index == " << imgIndex << endl;
if (imgIndex % (int)(allTotalFrame / 10000) == 0) { //假如取10000张,就是每 allTotalFrame/10000 帧截取一张
cout << "输出图片" << endl;
imwrite(imageSaveName, frame);
delete[] imageSaveName;
}
imgIndex++;
}
把视频按照规定的帧距离切割,并且保存切割出来的图片 全部代码:
#include <opencv2/opencv.hpp>
#include <opencv2/core/core.hpp>
#include <tchar.h>
#include <stdio.h>
#include <iostream>
#include <math.h>
#include <ostream>
#include <vector>
#include <string>
#include <stdlib.h>
#include <algorithm>
using namespace std;
using namespace cv;
vector<String> getFilePath() {
String dir_path = "F:\qietu\\video";
vector<String> videos_list; // opencv 2 String 改为 string即可
/*
函数名:glob
参数:String pattern 字符串,由文件夹路径和所要读取的文件名的格式构成的一个正则模板
参数: std::vector<String>& result 字符串容器,存放所有符合模板的路径
参数:bool recursive = false 标志位,是否使用递归的方式(没发现使用true和false的结果有什么区别)
功能描述:遍历指定的文件夹,读取符合搜索模板的所有文件路径
*/
glob(dir_path, videos_list,true);
for (size_t j = 0; j < videos_list.size(); ++j)
{
string fileName = videos_list[j];
}
return videos_list;
}
void main()
{
vector<String> listPath = getFilePath();
long allTotalFrame = 0;
for (int i = 0; i < listPath.size(); i++) {
VideoCapture cap;
cap.open(listPath[i]);
// 获取整视频帧数
long totalFrame = cap.get(CV_CAP_PROP_FRAME_COUNT);
allTotalFrame = allTotalFrame + totalFrame;
cout << "第 "<<i<<" 个视频宗贞数=" << totalFrame << endl;
}
cout << "总视频宗贞数=" << allTotalFrame << endl;
// 根据全部视频总帧数截取10000张图片,也就是每隔 allTotalFrame/10000 帧取一张
int imgIndex = 0;
for (int i = 0; i < listPath.size(); i++) {
VideoCapture cap;
cap.open(listPath[i]);
Mat frame;
cap >> frame;
for (;;)
{
Mat frame;
cap >> frame;
if (frame.empty())
{
break;
}
char* imageSaveName = new char[200];
//将图片的路径及名称输入到imageSaveName中
sprintf(imageSaveName, "F:\\test\\%05d.jpg", imgIndex);
cout << "index == " << imgIndex << endl;
if (imgIndex % (int)(allTotalFrame / 10000) == 0) { //假如取10000张,就是每 allTotalFrame/10000 帧截取一张
cout << "输出图片" << endl;
imwrite(imageSaveName, frame);
delete[] imageSaveName;
}
imgIndex++;
}
}
}