mat을 dlib의 행렬로 변환하는 방법

Question

192 채널의 행렬을 삽입해야하는 CNN을 사용하여 프로그램을 만드는 중입니다. 이 행렬 (폭 : 32, 높이 : 240, 채널 : 192, 유형 : uchar)은 Mat (OpenCV)에 저장됩니다. 매트를 dlib 행렬로 변환하는 방법은 무엇입니까?

std::vector<dlib::matrix<?>> training_data; 
std::vector<unsigned long> training_labels; 
... 
Mat mat = loader.getMat(); 
? convert ? 
training_data.push_back(dlib_matrix); 
... 
trainer.train(training_data, training_labels); 

Answer 1

당신은 이미지를 dlib하는 매트에서 변환 할 cv_image을 사용할 수 있으며 dlib::toMat는 매트 dlib 변환 할 수 있습니다.

//Mat to dlib image 
cv_image<bgr_pixel> dlib_img(mat); 

편집 :

는 지금까지 내가 알고, N 채널 매트에 대한 사용자 정의 pixel_traits을 제공했습니다. 예를 들어, 5 채널 매트 이미지를 당신이 할 수 있습니다 : 다음

namespace dlib{ 
struct custom_pixel 
    { 
     custom_pixel (
     ) {} 

     custom_pixel (
      unsigned char c1_, 
      unsigned char c2_, 
      unsigned char c3_, 
      unsigned char c4_, 
      unsigned char c5_ 
     ) : c1(c1_), c2(c2_), c3(c3_), c4(c4_), c5(c5_) {} 

     unsigned char c1; 
     unsigned char c2; 
     unsigned char c3; 
     unsigned char c4; 
     unsigned char c5; 
    }; 

template <> 
    struct pixel_traits<custom_pixel> 
    { 
     constexpr static bool rgb = false; 
     constexpr static bool rgb_alpha = false; 
     constexpr static bool grayscale = false; 
     constexpr static bool hsi = false; 
     constexpr static bool lab = false; 
     enum { num = 5};// provide number of channels here 
     typedef unsigned char basic_pixel_type; //provide channel depth here 
     static basic_pixel_type min() { return 0;} 
     static basic_pixel_type max() { return 255;} 
     constexpr static bool is_unsigned = true; 
     constexpr static bool has_alpha = false; 
    }; 
} 

는 dlib 그 반대에 매트로 변환하는 반대 :

int main(int argc, char** argv) 
{ 
    // from opencv to dlib 
    Mat mat_img = Mat::zeros(3, 3, CV_8UC(5)); 
    cv_image<custom_pixel> dlib_img(mat_img); 

    //from dlib to opencv 
    Mat mat_img_new = dlib::toMat(dlib_img); 
}