c++ - dlib-19.1:从图像(例如 dlib::cv_image)初始化 dlib::matrix 以进行 DNN 训练

Question

我目前正在尝试使用我存档的图像训练 DNN(OCR 上下文......每个类的输入图像是数千个固定大小的小图像的聚合图像)。

我有一些代码可以打开聚合图像并将其正确分割为小型 OpenCV cv::Mat。我的问题是，似乎没有办法

• 直接在 dlib::cv_image 上训练 DNN(它可以包裹在 cv::Mat 周围；我遇到了 500 多行编译器错误)或
• 无需复制每个元素即可轻松将 cv::Mat 转换/包装为 dlib::matrix

我很确定我在这里遗漏了一些东西，将不胜感激任何指点。

注意:我要编译的唯一变体是调用 dlib::dnn_trainer::train() 和一个 dlib::matrix vector (大小在编译时固定)和一个带有无符号长标签的 vector (无符号标签没有compile)，尽管 train() 在这两种类型上都是模板化的。有什么指点吗？

Answer 1

您不必在编译时固定 dlib::matrix 的大小。只需调用 set_size() 就可以了。另见 http://dlib.net/faq.html#HowdoIsetthesizeofamatrixatruntime .

另外，如果你想使用 dlib::matrix 以外的东西作为输入，你可以这样做。您只需要定义自己的输入层。您必须实现的接口(interface)已在此处完整记录:http://dlib.net/dlib/dnn/input_abstract.h.html#EXAMPLE_INPUT_LAYER .您还可以查看现有的输入层以获取示例。但请务必阅读文档，因为它会回答您可能遇到的问题。