2011-08-10 20 views
7

Estoy tratando de ejecutar kmeans una imagen en color de 3 canales en, pero cada vez que trato de ejecutar la función que parece chocar con el siguiente error:OpenCV algoritmo K-medias de funcionamiento en una imagen

OpenCV Error: Assertion failed (data.dims <= 2 && type == CV_32F && K > 0) in unknown function, file ..\..\..\OpenCV-2.3.0\modules\core\src\matrix.cpp, line 2271 

I' Hemos incluido el siguiente código con algunos comentarios para ayudar a especificar qué se está pasando. Cualquier ayuda es muy apreciada.

// Load in an image 
// Depth: 8, Channels: 3 
IplImage* iplImage = cvLoadImage("C:/TestImages/rainbox_box.jpg"); 

// Create a matrix to the image 
cv::Mat mImage = cv::Mat(iplImage); 

// Create a single channel image to create our labels needed 
IplImage* iplLabels = cvCreateImage(cvGetSize(iplImage), iplImage->depth, 1); 

// Convert the image to grayscale 
cvCvtColor(iplImage, iplLabels, CV_RGB2GRAY); 

// Create the matrix for the labels 
cv::Mat mLabels = cv::Mat(iplLabels); 

// Create the labels 
int rows = mLabels.total(); 
int cols = 1; 
cv::Mat list(rows, cols, mLabels .type()); 
uchar* src; 
uchar* dest = list.ptr(0); 
for(int i=0; i<mLabels.size().height; i++) 
{ 
    src = mLabels.ptr(i); 
    memcpy(dest, src, mLabels.step); 
    dest += mLabels.step; 
} 
list.convertTo(list, CV_32F); 

// Run the algorithm 
cv::Mat labellist(list.size(), CV_8UC1); 
cv::Mat centers(6, 1, mImage.type()); 
cv::TermCriteria termcrit(CV_TERMCRIT_EPS+CV_TERMCRIT_ITER, 10, 1.0); 
kmeans(mImage, 6, labellist, termcrit, 3, cv::KMEANS_PP_CENTERS, centers); 

Respuesta

11

El error dice todo: Assertion failed (data.dims <= 2 && type == CV_32F && K > 0)

Estas son reglas muy simples de entender, la función se activará sólo si:

  • mImage.depth() es CV_32F

  • si mImage.dims es <= 2

  • y si K > 0. En este caso, defina K como .

De lo que se indica en la pregunta, parece que:

IplImage* iplImage = cvLoadImage("C:/TestImages/rainbox_box.jpg");` 

se está cargando la imagen como IPL_DEPTH_8U por defecto y no IPL_DEPTH_32F. Esto significa que mImage también es IPL_DEPTH_8U, por lo que su código no funciona.

+0

¿Significa esto que tengo que cargarlo como una imagen en escala de grises? – Seb

+0

No, no es así. 'iplImage-> depth' y' iplImage-> nChannels' son dos cosas diferentes. Necesita convertir iplImage a su equivalente IPL_DEPTH_32F. Eso es todo. Verifique esta respuesta sobre cómo hacer eso: http://stackoverflow.com/questions/6349493/opencv-iplimage-data-to-float – karlphillip

+0

Acabo de probar la función de conversión y se cuelga en mí. "IplImage * ipl32Image = cvCreateImage (cvGetSize (iplImage), IPL_DEPTH_32F, 1); cvConvert (iplImage, ipl32Image);" con el error "Error de OpenCV: la aserción falló (src.size == dst.size && src.channels() == dst.ch annels()) en la función desconocida, archivo ... \ .. \ .. \ OpenCV-2.3 .0 \ modules \ core \ src \ convert.cpp, línea 1177 ". Parece que tendré que dividir los canales o tendré que cargar una imagen en escala de grises – Seb

Cuestiones relacionadas