examples:classification: Use configuration.numFrames instead of hard coded big value

Signed-off-by: Djordje Senicic <x0157990@ti.com>

diff --git a/examples/classification/main.cpp b/examples/classification/main.cpp
index 99bf7b202acbec68251113635f017994afa46138..21c05a74f4ac39d0a81703edc179b580a728433c 100644 (file)
--- a/examples/classification/main.cpp
+++ b/examples/classification/main.cpp
@@ -230,7 +230,7 @@ bool RunConfiguration(const std::string& config_file, int num_layers_groups, uin
             for(int i = 0; i < 3; i ++)
                 selclass_history[k][i] = -1;
         avg_fps = 0.0;
             for(int i = 0; i < 3; i ++)
                 selclass_history[k][i] = -1;
         avg_fps = 0.0;

-        int num_frames = 999999;
+        int num_frames = configuration.numFrames;

         std::cout << "About to start ProcessFrame loop!!" << std::endl;
  
         // Process frames with available EOPs in a pipelined manner
         std::cout << "About to start ProcessFrame loop!!" << std::endl;
  
         // Process frames with available EOPs in a pipelined manner


@@ -609,8 +609,6 @@ void DisplayFrame(const ExecutionObjectPipeline* eop, VideoWriter& writer,
     cv::imshow(imagenet_win, show_image);
 #endif
 
     cv::imshow(imagenet_win, show_image);
 #endif
 

-#endif
-

 #ifdef RMT_GST_STREAMER
     cv::resize(show_image, to_stream, cv::Size(640,480));
     writer << to_stream;
 #ifdef RMT_GST_STREAMER
     cv::resize(show_image, to_stream, cv::Size(640,480));
     writer << to_stream;