如何将cv::Mat转换为torch::张量并将其馈送到libtorch模型?

Question

我用cv2.imread()读取图像，并试图将其提供给c++中的火炬模型。它的数据类型cv::Mat。我想我需要将其转换为张量，然后使用model.forward()，但我很困惑如何做到这一点。python中是否有类似于。tensor()的函数?

Answer 1

函数torch::from_blob可用于在图像数据上创建张量视图，如下所示:

torch::Tensor to_tensor(cv::Mat img) {
return torch::from_blob(img.data, { img.rows, img.cols, 3 }, torch::kUInt8);
}