将图像从URL加载到变量/ opencv Mat中

时间:2015-03-21 00:14:27

标签: c++ opencv curl opencv-mat

bool loadImage(string inputName, Mat &image)
{
  bool from_net = true;

  if (inputName.find("http") != string::npos)
    {
      string URL = inputName;
      if (inputName.find("\"") == 0)
        {
          URL = inputName.substr(1,inputName.length()-2);
        }
      ofstream myfile;
      myfile.open ("test.jpg");
      //  Create a writer to handle the stream

      curl_writer writer(myfile);
      // Pass it to the easy constructor and watch the content returned in that file!
      curl_easy easy(writer);

      // Add some option to the easy handle
      easy.add(curl_pair<CURLoption,string>(CURLOPT_URL,URL));
      easy.add(curl_pair<CURLoption,long>(CURLOPT_FOLLOWLOCATION,1L));
      try {
        easy.perform();
      } catch (curl_easy_exception error) {
        // If you want to get the entire error stack we can do:
        vector<pair<string,string>> errors = error.what();
        // Otherwise we could print the stack like this:
        //error.print_traceback();
      }
      myfile.close();

      string inputname = "test.jpg";
      image = imread(inputname,1);

      if(image.rows == 0 || image.cols == 0)
          from_net = false;
    }  
  else
    {
      image = imread( inputName, 1 );
      if (image.total() < 1)         
          from_net = false;

    }
  return from_net; 
}

如果我将test.txt更改为test.jpg,这适用于我的应用程序。但是,我的应用程序要求我避免创建文件,读取,写入和关闭文件的开销。是否有一种简单直接的方法从URL获取图像数据并将其写入openCV Mat?

我也尝试了上面链接中的第3个例子。但由于某些原因,当我执行receiver.get_buffer()时,图像仍为空。我将图片尺寸设为0X0

任何与此相关的帮助都非常感谢。我之前从未使用过curlcpp,因此,对此相关的任何详细解释都将非常感激。

感谢。

1 个答案:

答案 0 :(得分:1)

有一个简单的解决方案,我不喜欢早先注意到这一点。您可以将数据写入ostringstream。有关详细信息,请参阅以下代码。

bool loadImage(string inputName, Mat &image)
{
  bool from_net;
  from_net = true;


  if (inputName.find("http") != string::npos)
    {
      string URL;
      URL = inputName;
      if (inputName.find("\"") == 0)
        {
           URL = inputName.substr(1,inputName.length()-2);
        }

  std::ostringstream stream;

  curl_writer writer(stream);
  // Pass it to the easy constructor and watch the content returned in that file!
  curl_easy easy(writer);

  // Add some option to the easy handle
  easy.add(curl_pair<CURLoption,string>(CURLOPT_URL,URL));
  easy.add(curl_pair<CURLoption,long>(CURLOPT_FOLLOWLOCATION,1L));

  try {
    easy.perform();
  } catch (curl_easy_exception error) {
    // If you want to get the entire error stack we can do:
    vector<pair<string,string>> errors = error.what();
    // Otherwise we could print the stack like this:
    error.print_traceback();
  }
  string output = stream.str(); // convert the stream into a string
  if (output.find("404 Not Found") != string::npos)
    from_net = false;
  else
  {
      vector<char> data = std::vector<char>( output.begin(), output.end() ); //convert string into a vector 
  if (data.size() > 0)
    {
      Mat data_mat = Mat(data); // create the cv::Mat datatype from the vector
      image = imdecode(data_mat,-1); //read an image from memory buffer
      if(image.rows == 0 || image.cols == 0)
    from_net = false;
    }
  else
    from_net = false;
    }
}
  else
    {
      image = imread( inputName, 1 );
      if (image.total() < 1)         
        from_net = false;   
    }
  return from_net;
}