c++ - SLIC c++ 分段

Question

我试图在 OpenCV 中使用 SLIC 分割图像。我正在尝试使用以下功能:

void vl_slic_segment    (   vl_uint32 *     segmentation,
float const *   image,
vl_size     width,
vl_size     height,
vl_size     numChannels,
vl_size     regionSize,
float   regularization,
vl_size     minRegionSize 
)

#include 很好，链接到库也很好。我只需要知道如何将图像传递给此函数。此函数中的图像参数是 float const * 类型，我不知道如何将图像转换为这种类型。

下面是我如何将图像加载到代码中:

IplImage *image = cvLoadImage("train.tif", 1);

这里是完整的代码:

extern "C" {
  #include </home/me/Downloads/vlfeat-0.9.17/vl/slic.h>
}
#include <stdio.h>
#include <iostream>
#include <string>
#include <opencv2/opencv.hpp>
#include<opencv/highgui.h>

using namespace std;
using namespace cv;

int main () {
    IplImage *image = cvLoadImage("train.tif", 1);

   vl_uint32 * seg;

   vl_slic_segment(seg,(const float *)image,image->width,image->height,image->nChannels,15,0.1,1);

  waitKey(0);
}

而且我不知道我是否正确使用了 vl_uint32 * seg。如果有人有示例或示例代码来执行此 segmentation ，请。

谢谢!!

Answer 1

您需要为seg 正确分配存储空间。如果您打算像 berak 的回答中那样使用 C++ API(我也建议这样做)，您可以创建一个 Mat 来保存标签数据，以便以后更轻松地访问并自动管理内存:

cv::Mat labels(floatimg.size(), CV_32SC1); // Mat doesn't support 32-bit unsigned directly, but this should work fine just to hold data.
vl_slic_segment(labels.ptr<vl_uint32>(),floatimg.ptr<float>(),floatimg.cols,floatimg.rows,floatimg.channels(),15,0.1,1);

如果出于某种原因您不想这样做，您可以像这样分配一 block 原始内存(不推荐):

vl_uint32* seg = new vl_uint32[floatimg.total()]; // don't forget to delete[]

或者如果您决定继续使用 C API，您可以使用 malloc(真的不推荐):

vl_uint32* seg = (vl_uint32*)malloc(img->height * img->width); // don't forget to free()