mmr/tkDNN
Archived
1
0
Fork 0
Code Issues 46 Pull Requests 12 Releases 4 Activity

save video result

Browse Source
This commit is contained in:
Francesco Gatti
2019-09-16 10:35:29 +02:00
parent a038e966d9
commit 77f031c0f4
1 changed files with 12 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
demo/demo/demo.cpp
+12 -1
View File
@@ -16,6 +16,7 @@ bool gRun;
//std::string obj_class[3] {"person", "bike", "car"}; //std::string obj_class[3] {"person", "bike", "car"};
std::string obj_class[10] {"0", "1", "2", "3", "4", "5", "6", "7", "8", "9"}; std::string obj_class[10] {"0", "1", "2", "3", "4", "5", "6", "7", "8", "9"};
bool SAVE_RESULT = false;
void sig_handler(int signo) { void sig_handler(int signo) {
std::cout<<"request gateway stop\n"; std::cout<<"request gateway stop\n";
@@ -46,6 +47,14 @@ int main(int argc, char *argv[]) {
else else
std::cout<<"camera started\n"; std::cout<<"camera started\n";
cv::VideoWriter resultVideo;
if(SAVE_RESULT) {
int w = cap.get(CV_CAP_PROP_FRAME_WIDTH);
int h = cap.get(CV_CAP_PROP_FRAME_HEIGHT);
resultVideo.open("result.mp4", CV_FOURCC('M','P','4','V'), 30, cv::Size(w, h));
}
cv::Mat frame; cv::Mat frame;
cv::Mat dnn_input; cv::Mat dnn_input;
cv::namedWindow("detection", cv::WINDOW_NORMAL); cv::namedWindow("detection", cv::WINDOW_NORMAL);
@@ -53,7 +62,7 @@ int main(int argc, char *argv[]) {
while(gRun) { while(gRun) {
cap >> frame; cap >> frame;
if(!frame.data) { if(!frame.data) {
continue; break;
} }
// this will be resized to the net format // this will be resized to the net format
@@ -86,6 +95,8 @@ int main(int argc, char *argv[]) {
cv::imshow("detection", frame); cv::imshow("detection", frame);
cv::waitKey(1); cv::waitKey(1);
if(SAVE_RESULT)
resultVideo << frame;
} }
std::cout<<"detection end\n"; std::cout<<"detection end\n";