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

Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com>
2022-01-19 10:22:06 -08:00
parent 907df27e07
commit decd73d298
3 changed files with 16 additions and 5 deletions
@@ -30,6 +30,9 @@ class DepthNN {
        dnnType *input_d;
        float* depth_h;

+        int output_w;
+        int output_h;
+
        int nBatches = 1;

        cv::Mat bgr[3];
@@ -68,6 +71,9 @@ class DepthNN {
                depths[i].resize(netRT->buffersDIM[1].tot());

            depth_h = (float *)malloc(netRT->buffersDIM[1].tot() * sizeof(float));
+
+            output_h = netRT->buffersDIM[1].h;
+            output_w = netRT->buffersDIM[1].w;
            
        }