效果:
或者灰度,cell大小可调
代码:
#include <opencv2opencv.hpp> #include <Windows.h> struct parameters { parameters() { size=cv::Size(640,480); fps=25; nF=50; save=""; crossSize= cv::Size(15, 5); crossColor = 0; show=true; seed=cv::getTickCount(); color=false; cellSize = cv::Size(5, 5); } cv::Size size; int fps; int nF; std::string save; cv::Size crossSize; uchar crossColor; bool show; int64 seed; bool color; cv::Size cellSize; void setParas( cv::Size _size, int _fps, int _nF, std::string _save, cv::Size _crossSize, uchar _crossColor, bool _show, int64 _seed, bool _color, cv::Size _cellSize) { size = _size ; fps = _fps ; nF = _nF ; save = _save ; crossSize = _crossSize ; crossColor = _crossColor ; show = _show ; seed = _seed ; color = _color ; cellSize = _cellSize ; } void showParas() { std::cout << "size: " << size << std::endl; std::cout << "fps: " << fps << std::endl; std::cout << "nF: " << nF << std::endl; std::cout << "save: " << save << std::endl; std::cout << "crossSize: " << crossSize << std::endl; std::cout << "crossColor: " << crossColor << std::endl; std::cout << "show: " << (show? "TRUE" : "FALSE") << std::endl; std::cout << "seed: " << seed << std::endl; std::cout << "color: " << color << std::endl; std::cout << "cellSize: " << cellSize << std::endl; } }; void help() { std::cout << "* -help : . " << " randomVideo.exe -help | to see how to use. " << " randomVideo.exe -size=(640,580) | to set video size. " << " randomVideo.exe -fps=25 | to set video fps(default is 25). " << " randomVideo.exe -nF=50 | to set number of frames. " << "If fps is 25, 50 nF means the video will last 2 seconds (50/25). " << " randomVideo.exe -save=example1 | " << "to set video name, only *.avi support, " << "you do not need input ".avi". " << "If empty, the default name is local-system-time.avi, your previous files are safe. " << " randomVideo.exe -crossSize=(15,5) | " << "to set the size of cross." << " (height, width) you can see the showing window. " << " randomVideo.exe -crossColor=0 | Recommd not to set it. " << "If = 0, black(default); if = 255, white. " << " randomVideo.exe -show=TRUE | Recommd to be TRUE. " <<"If FALSE, the window to show the video is stopped. " << " randomVideo.exe -seed=15 | Recommd not to set it. " <<"If empty, the local time will be used, then," <<" you will get different video each time. " << " { NEW } randomVideo.exe -color=TRUE | If FALSE, the video is gray. " << " { NEW } randomVideo.exe -cellSize=(2,3) | To control the size of pixel size. " << "If cellSize=(2,3), the image width will be seperated into 2 parts, " << "the image height will be seperated into 3 parts. In defalut mode pixel size as 1 will be used!" << "In order to get exact pixel size, you need to calculate it by yourself. " << "example: " << " randomVideo.exe -size=(640,580) -fps=25 -nF=50 -crossSize=(15,5) -color=1 -cellSize=(5,5) " << " The other parameters using default is fine. " << std::endl; } void getParas(int argn, char** argv, parameters & _parameters) { if (argn <= 1) { std::cout << "Using default parameter to generate video. " << "How to use? Please check the usage by input " randomVideo.exe -help" " << std::endl; help(); exit(0); } else { std::map<std::string, std::string> table; for (int i = 1; i<argn; ++i) { std::string sub(argv[i]); std::string name = sub.substr(1, sub.find("=")-1); table.insert({name, sub.substr(sub.find("=")+1, sub.length()-sub.find("=")-1)}); } if (table.count("help") == 1) { help(); exit(0); } if (table.count("size") == 1) { cv::Size _size(atoi(table["size"].substr(1, table["size"].find(",")-1).c_str()), atoi(table["size"].substr( table["size"].find(",")+1, table["size"].length()-table["size"].find(",")-2 ).c_str())); _parameters.size = _size; } else _parameters.size = cv::Size(640, 480); if (table.count("crossSize") == 1) { cv::Size _crossSize(atoi(table["crossSize"].substr(1, table["crossSize"].find(",") - 1).c_str()), atoi(table["crossSize"].substr( table["crossSize"].find(",") + 1, table["crossSize"].length() - table["crossSize"].find(",") - 2 ).c_str())); _parameters.crossSize = _crossSize; } else _parameters.crossSize = cv::Size(15, 5); if (table.count("crossColor") == 1) _parameters.crossColor = atoi(table["crossColor"].c_str()); else _parameters.crossColor = 0; if (table.count("fps") == 1) _parameters.fps = atoi(table["fps"].c_str()); else _parameters.fps = 25; if (table.count("nF") == 1) _parameters.nF = atoi(table["nF"].c_str()); else _parameters.nF = 50; if (table.count("save") == 1) _parameters.save = table["save"] + ".avi"; else { SYSTEMTIME sys; GetLocalTime(&sys); char fname[100] = {}; sprintf_s(fname, 100, "%4d-%02d-%02d_%02d_%02d_%02d.avi", sys.wYear, sys.wMonth, sys.wDay, sys.wHour, sys.wMinute, sys.wSecond); _parameters.save = fname; } if (table.count("show") == 1) _parameters.show = (table["show"] == "TRUE"); else _parameters.show = true; if (table.count("seed") == 1) _parameters.seed = atoi(table["seed"].c_str()); else { SYSTEMTIME sys; GetLocalTime(&sys); _parameters.seed = sys.wYear + sys.wMonth + sys.wDay + sys.wHour + sys.wMinute + sys.wSecond; } /*if (table.count("color") == 1) { cv::Scalar _color( atoi(table["color"].substr(1, table["color"].find(",") - 1).c_str()), atoi(table["color"].substr(table["color"].find(",") + 1, table["color"].find_last_of(",") - table["color"].find(",") - 2).c_str()), atoi(table["color"].substr(table["color"].find_last_of(",") + 1, table["color"].find_last_of(")") - table["color"].find_last_of(",") - 1).c_str())); _parameters.color = _color; } else _parameters.size = cv::Size(640, 480);*/ if (table.count("color") == 1) _parameters.color = (table["color"] == "TRUE"); else _parameters.color = true; if (table.count("cellSize") == 1) { cv::Size _cellSize(atoi(table["cellSize"].substr(1, table["cellSize"].find(",") - 1).c_str()), atoi(table["cellSize"].substr( table["cellSize"].find(",") + 1, table["cellSize"].length() - table["cellSize"].find(",") - 2 ).c_str())); _parameters.cellSize = _cellSize; } else _parameters.cellSize = cv::Size(2, 2); } _parameters.showParas(); } void randMatrix(cv::Mat & img, int64 seed, cv::Size cell=cv::Size(0,0)) { cv::RNG rng(seed); if (cell == cv::Size(0, 0) || cell.width > img.cols || cell.height > img.rows) { for (size_t j = 0; j < img.rows; ++j) { uchar* pdata = img.ptr<uchar>(j); for (size_t i = 0; i < img.cols; ++i) { pdata[i] = rng.uniform(0, 255); } } } else { size_t stepX = img.cols / cell.width; size_t stepY = img.rows / cell.height; for (size_t j = 0; j < img.rows; j+=stepY) { size_t lastY(j + stepY); if (lastY >= img.rows) lastY = img.rows - 1; for (size_t i = 0; i < img.cols; i += stepX) { size_t lastX(i+stepX); if (lastX >= img.cols) lastX = img.cols - 1; img.rowRange(j, lastY).colRange(i, lastX).setTo(rng.uniform(0,255)); } } } } void addCross(cv::Mat & img, cv::Size cSzie, cv::Scalar scalar) { int x_center = (img.cols - 1) / 2, y_center = (img.rows - 1) / 2; cv::Point P1_tl(x_center - (cSzie.width - 1) / 2, y_center - (cSzie.height - 1) / 2), P1_bd(x_center + (cSzie.width - 1) / 2, y_center + (cSzie.height - 1) / 2), P2_tl(x_center - (cSzie.height - 1) / 2, y_center - (cSzie.width - 1) / 2), P2_bd(x_center + (cSzie.height - 1) / 2, y_center + (cSzie.width - 1) / 2); cv::rectangle(img, P1_tl, P1_bd, scalar, -1); cv::rectangle(img, P2_tl, P2_bd, scalar, -1); } int main(int argn, char** argv) { parameters _pa; getParas(argn, argv, _pa); cv::VideoWriter writer(_pa.save, CV_FOURCC('M', 'J','P','G'), double(_pa.fps), _pa.size); cv::Mat frame0 = cv::Mat::zeros(_pa.size, CV_8UC1); cv::Mat frame1 = cv::Mat::zeros(_pa.size, CV_8UC1); cv::Mat frame2 = cv::Mat::zeros(_pa.size, CV_8UC1); for (int i=0; i<_pa.nF; ++i) { randMatrix(frame0, _pa.seed + i, _pa.cellSize); addCross(frame0, _pa.crossSize, _pa.crossColor); randMatrix(frame1, 0.3*(_pa.seed + i), _pa.cellSize); addCross(frame1, _pa.crossSize, _pa.crossColor); randMatrix(frame2, 0.7*(_pa.seed + i), _pa.cellSize); addCross(frame2, _pa.crossSize, _pa.crossColor); cv::Mat f3; if(_pa.color) cv::merge(std::vector<cv::Mat>{frame0,frame1,frame2}, f3); else cv::merge(std::vector<cv::Mat>{frame0, frame0, frame0}, f3); writer << f3; if (_pa.show) cv::imshow("video", f3); cv::waitKey(10); } return 0; }
使用说明:
见help()