samples/cpp/stitching.cpp

   1 /*M///////////////////////////////////////////////////////////////////////////////////////
   2 //
   3 //  IMPORTANT: READ BEFORE DOWNLOADING, COPYING, INSTALLING OR USING.
   4 //
   5 //  By downloading, copying, installing or using the software you agree to this license.
   6 //  If you do not agree to this license, do not download, install,
   7 //  copy or use the software.
   8 //
   9 //
  10 //                          License Agreement
  11 //                For Open Source Computer Vision Library
  12 //
  13 // Copyright (C) 2000-2008, Intel Corporation, all rights reserved.
  14 // Copyright (C) 2009, Willow Garage Inc., all rights reserved.
  15 // Third party copyrights are property of their respective owners.
  16 //
  17 // Redistribution and use in source and binary forms, with or without modification,
  18 // are permitted provided that the following conditions are met:
  19 //
  20 //   * Redistribution's of source code must retain the above copyright notice,
  21 //     this list of conditions and the following disclaimer.
  22 //
  23 //   * Redistribution's in binary form must reproduce the above copyright notice,
  24 //     this list of conditions and the following disclaimer in the documentation
  25 //     and/or other materials provided with the distribution.
  26 //
  27 //   * The name of the copyright holders may not be used to endorse or promote products
  28 //     derived from this software without specific prior written permission.
  29 //
  30 // This software is provided by the copyright holders and contributors "as is" and
  31 // any express or implied warranties, including, but not limited to, the implied
  32 // warranties of merchantability and fitness for a particular purpose are disclaimed.
  33 // In no event shall the Intel Corporation or contributors be liable for any direct,
  34 // indirect, incidental, special, exemplary, or consequential damages
  35 // (including, but not limited to, procurement of substitute goods or services;
  36 // loss of use, data, or profits; or business interruption) however caused
  37 // and on any theory of liability, whether in contract, strict liability,
  38 // or tort (including negligence or otherwise) arising in any way out of
  39 // the use of this software, even if advised of the possibility of such damage.
  40 //
  41 //M*/
  42
  43 #include <iostream>
  44 #include <fstream>
  45 #include "opencv2/imgcodecs.hpp"
  46 #include "opencv2/highgui.hpp"
  47 #include "opencv2/stitching.hpp"
  48
  49 using namespace std;
  50 using namespace cv;
  51
  52 bool try_use_gpu = false;
  53 vector<Mat> imgs;
  54 string result_name = "result.jpg";
  55
  56 void printUsage();
  57 int parseCmdArgs(int argc, char** argv);
  58
  59 int main(int argc, char* argv[])
  60 {
  61     int retval = parseCmdArgs(argc, argv);
  62     if (retval) return -1;
  63
  64     Mat pano;
  65     Stitcher stitcher = Stitcher::createDefault(try_use_gpu);
  66     Stitcher::Status status = stitcher.stitch(imgs, pano);
  67
  68     if (status != Stitcher::OK)
  69     {
  70         cout << "Can't stitch images, error code = " << int(status) << endl;
  71         return -1;
  72     }
  73
  74     imwrite(result_name, pano);
  75     return 0;
  76 }
  77
  78
  79 void printUsage()
  80 {
  81     cout <<
  82         "Rotation model images stitcher.\n\n"
  83         "stitching img1 img2 [...imgN]\n\n"
  84         "Flags:\n"
  85         "  --try_use_gpu (yes|no)\n"
  86         "      Try to use GPU. The default value is 'no'. All default values\n"
  87         "      are for CPU mode.\n"
  88         "  --output <result_img>\n"
  89         "      The default is 'result.jpg'.\n";
  90 }
  91
  92
  93 int parseCmdArgs(int argc, char** argv)
  94 {
  95     if (argc == 1)
  96     {
  97         printUsage();
  98         return -1;
  99     }
 100     for (int i = 1; i < argc; ++i)
 101     {
 102         if (string(argv[i]) == "--help" || string(argv[i]) == "/?")
 103         {
 104             printUsage();
 105             return -1;
 106         }
 107         else if (string(argv[i]) == "--try_use_gpu")
 108         {
 109             if (string(argv[i + 1]) == "no")
 110                 try_use_gpu = false;
 111             else if (string(argv[i + 1]) == "yes")
 112                 try_use_gpu = true;
 113             else
 114             {
 115                 cout << "Bad --try_use_gpu flag value\n";
 116                 return -1;
 117             }
 118             i++;
 119         }
 120         else if (string(argv[i]) == "--output")
 121         {
 122             result_name = argv[i + 1];
 123             i++;
 124         }
 125         else
 126         {
 127             Mat img = imread(argv[i]);
 128             if (img.empty())
 129             {
 130                 cout << "Can't read image '" << argv[i] << "'\n";
 131                 return -1;
 132             }
 133             imgs.push_back(img);
 134         }
 135     }
 136     return 0;
 137 }