2011-08-09 16 views
13

Quiero copiar las filas 0, 2 y 4 de mi matriz A en B, en este orden. Deje A = [a0, a1, a2, a3, a4]^T, con a_i siendo vectores de fila, , entonces B debería ser: [a0, a2, a4]^T.Copiar seleccionar filas en la nueva matriz

El siguiente código hace lo que quiero pero me pregunto si existe una solución más bonita (¿tal vez usando Eigen)?

#include <iostream> 
#include <vector> 
#include <opencv/cv.h> 


int main(int argc, char **argv) { 
    const int num_points = 5; 
    const int vec_length = 3; 
    cv::Mat A(num_points, vec_length, CV_32FC1); 
    cv::RNG rng(0); // Fill A with random values 
    rng.fill(A, cv::RNG::UNIFORM, 0, 1); 
// HACK Ugly way to fill that matrix . 
    cv::Mat B = cv::Mat(3,vec_length, CV_32FC1); 
    cv::Mat tmp0 = B(cv::Rect(0,0,vec_length,1)); 
    cv::Mat tmp1 = B(cv::Rect(0,1,vec_length,1)); 
    cv::Mat tmp2 = B(cv::Rect(0,2,vec_length,1)); 
    A.row(0).copyTo(tmp0); 
    A.row(2).copyTo(tmp1); 
    A.row(4).copyTo(tmp2); 

    std::cout << "A: " << A << std::endl; 
    std::cout << "B: " << B << std::endl; 
    return 0; 
} 

Respuesta

17

He encontrado push_back.

Crear B con size 0 x vec_length y luego usar push_back añadir las filas seleccionadas de A:

#include <iostream> 
#include <vector> 
#include <opencv/cv.h> 


int main(int argc, char **argv) { 
    const int num_points = 5; 
    const int vec_length = 3; 
    cv::Mat A(num_points, vec_length, CV_32FC1); 
    cv::RNG rng(0); // Fill A with random values 
    rng.fill(A, cv::RNG::UNIFORM, 0, 1); 
    cv::Mat B = cv::Mat(0,vec_length, CV_32FC1); 
    B.push_back(A.row(0)); 
    B.push_back(A.row(2)); 
    B.push_back(A.row(4)); 
    std::cout << "A: " << A << std::endl; 
    std::cout << "B: " << B << std::endl; 
    return 0; 
} 
+0

¡esto es asombroso! ¡Tratando cv :: Mat como vectores! Gracias a la nueva API de C++. – garak

1

Como no son contiguas, no creo que haya ningún atajo. En este caso particular, se podría hacer el código más limpio con un bucle:

for (int i=0; i<3; i++) { 
    cv::Mat tmp = B(cv::Rect(0,i,vec_length,1)); 
    A.row(i * 2).copyTo(tmp); 
} 
+0

bien, pero en mi caso de que el acceso directo no funcione porque estoy usando valores diferentes a (0,2,4). ¡Gracias de cualquier manera! – Unapiedra

+0

Dependiendo de su circunstancia, tal vez podría crear un vector de duplicados de origen y destino. Puede usar un bucle similar que se ejecuta a través del vector y copia la fila de origen a la fila de destino. ¡Buena suerte! – SSteve

23

La forma en OpenCV 2.4.1 es:

A.row(0).copyTo(B.row(0)); 
A.row(2).copyTo(B.row(1)); 
A.row(4).copyTo(B.row(2)); 
0
Mat row = old.row(0); 
old.row(0).copyTo(row); 
new.push_back(row); 
Cuestiones relacionadas