examples:classification: Use configuration.numFrames instead of hard coded big value
authorDjordje Senicic <x0157990@ti.com>
Fri, 7 Sep 2018 17:34:37 +0000 (13:34 -0400)
committerDjordje Senicic <x0157990@ti.com>
Fri, 7 Sep 2018 17:34:37 +0000 (13:34 -0400)
Signed-off-by: Djordje Senicic <x0157990@ti.com>
examples/classification/main.cpp

index 99bf7b202acbec68251113635f017994afa46138..21c05a74f4ac39d0a81703edc179b580a728433c 100644 (file)
@@ -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;
-        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
@@ -609,8 +609,6 @@ void DisplayFrame(const ExecutionObjectPipeline* eop, VideoWriter& writer,
     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;