update structured forests tutorial

modules/ximgproc/samples/structured_edge_detection.cpp

@@ -1,12 +1,11 @@
 /**************************************************************************************
-The structered edge demo requires you to provide a model.
+The structured forests for fast edge detection demo requires you to provide a model.
 This model can be found at the opencv_extra repository on Github on the following link:
 https://github.com/opencv/opencv_extra/blob/master/testdata/cv/ximgproc/model.yml.gz
 ***************************************************************************************/
 
 #include <opencv2/ximgproc.hpp>
 #include "opencv2/highgui.hpp"
-#include "opencv2/core/utility.hpp"
 #include <iostream>
 
 using namespace cv;
@@ -14,26 +13,17 @@ using namespace cv::ximgproc;
 
 const char* keys =
 {
-    "{i || input image name}"
-    "{m || model name}"
-    "{o || output image name}"
+    "{i || input image file name}"
+    "{m || model file name}"
+    "{o || output image file name}"
 };
 
 int main( int argc, const char** argv )
 {
-    bool printHelp = ( argc == 1 );
-    printHelp = printHelp || ( argc == 2 && String(argv[1]) == "--help" );
-    printHelp = printHelp || ( argc == 2 && String(argv[1]) == "-h" );
-
-    if ( printHelp )
-    {
-        std::cout << "\nThis sample demonstrates structured forests for fast edge detection\n"
-               "Call:\n"
-               "    structured_edge_detection -i=in_image_name -m=model_name [-o=out_image_name]\n\n";
-        return 0;
-    }
-
     CommandLineParser parser(argc, argv, keys);
+    parser.about("This sample demonstrates usage of structured forests for fast edge detection");
+    parser.printMessage();
+
     if ( !parser.check() )
     {
         parser.printErrors();
@@ -44,21 +34,41 @@ int main( int argc, const char** argv )
     String inFilename = parser.get<String>("i");
     String outFilename = parser.get<String>("o");
 
-    Mat image = imread(inFilename, 1);
+    //! [imread]
+    Mat image = imread(inFilename, IMREAD_COLOR);
     if ( image.empty() )
         CV_Error(Error::StsError, String("Cannot read image file: ") + inFilename);
+    //! [imread]
 
     if ( modelFilename.size() == 0)
         CV_Error(Error::StsError, String("Empty model name"));
 
+    //! [convert]
     image.convertTo(image, DataType<float>::type, 1/255.0);
+    //! [convert]
 
-    Mat edges(image.size(), image.type());
-
+    TickMeter tm;
+    tm.start();
+    //! [create]
     Ptr<StructuredEdgeDetection> pDollar =
         createStructuredEdgeDetection(modelFilename);
-    pDollar->detectEdges(image, edges);
+    //! [create]
 
+    tm.stop();
+    std::cout << "createStructuredEdgeDetection() time : " << tm << std::endl;
+
+    tm.reset();
+    tm.start();
+    //! [detect]
+    Mat edges;
+    pDollar->detectEdges(image, edges);
+    //! [detect]
+    tm.stop();
+    std::cout << "detectEdges() time : " << tm << std::endl;
+
+    tm.reset();
+    tm.start();
+    //! [nms]
     // computes orientation from edge map
     Mat orientation_map;
     pDollar->computeOrientation(edges, orientation_map);
@@ -66,17 +76,21 @@ int main( int argc, const char** argv )
     // suppress edges
     Mat edge_nms;
     pDollar->edgesNms(edges, orientation_map, edge_nms, 2, 0, 1, true);
+    //! [nms]
 
+    tm.stop();
+    std::cout << "nms time : " << tm << std::endl;
+
+    //! [imshow]
     if ( outFilename.size() == 0 )
     {
-        namedWindow("edges", 1);
         imshow("edges", edges);
-        namedWindow("edges nms", 1);
         imshow("edges nms", edge_nms);
         waitKey(0);
     }
     else
         imwrite(outFilename, 255*edges);
+    //! [imshow]
 
     return 0;
 }