0

我使用 CNN 创建程序，我需要在其中插入 192 个通道的矩阵。这个矩阵（宽度：32，高度：240，通道：192，类型：uchar）存储在 Mat（OpenCV）中。如何将 Mat 转换为 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);
4

1 回答 1

1

您可以使用cv_image将 Mat 转换为 dlib 图像，使用 dlib::toMat将 dlib 转换为 Mat。

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

编辑：

据我所知，对于 n 通道 Mats，您必须提供 custom pixel_traits. 例如，对于 5 通道 Mat 图像，您可以执行以下操作：

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;
    };
}

然后从 Mat 转换为 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);
}
于 2017-10-28T12:21:37.000 回答