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

Fix saving result video in demoDepth, add automatic download for monodepth2 weights

Browse Source 
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com>
This commit is contained in:
Micaela Verucchi
2022-01-19 10:22:06 -08:00
parent 907df27e07
commit decd73d298
3 changed files with 16 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files
demo/demo/demoDepth.cpp
+6 -3
View File
@@ -55,9 +55,9 @@ int main(int argc, char *argv[]) {
cv::VideoWriter resultVideo;
if(save) {
int w = cap.get(cv::CAP_PROP_FRAME_WIDTH);
int h = cap.get(cv::CAP_PROP_FRAME_HEIGHT);
resultVideo.open("result.mp4", cv::VideoWriter::fourcc('M','P','4','V'), 30, cv::Size(w, h));
int w = depthNN.output_w;
int h = depthNN.output_h;
resultVideo.open("result.mp4", cv::VideoWriter::fourcc('M','J','P','G'), 30, cv::Size(w, h));
}
if(show)
@@ -87,6 +87,9 @@ int main(int argc, char *argv[]) {
cv::waitKey(1);
}
if(save)
resultVideo << depthNN.depthMats[0];
}
std::cout<<"detection end\n";