tests: Run prediction 10 times

ethz-asl · Nov 11, 2024 · 34f32aa · 34f32aa
1 parent 8c02c34
commit 34f32aa
Showing 1 changed file with 7 additions and 1 deletion.
diff --git a/test/test_depth_anything_v2.cpp b/test/test_depth_anything_v2.cpp
@@ -59,5 +59,11 @@ TEST_F(TestDepthAnythingV2, TestSetInput) {
 }
 
 TEST_F(TestDepthAnythingV2, TestPredict) {
-    depth_anything_v2->predict();
+    auto start = std::chrono::high_resolution_clock::now();
+    for (size_t i = 0; i < 10; i++) {
+        depth_anything_v2->predict();
+    }
+    auto end = std::chrono::high_resolution_clock::now();
+    std::chrono::duration<double> elapsed = end - start;
+    std::cout << "Time taken for 10 predictions: " << elapsed.count() << " seconds." << std::endl;
 }