如何將 cv::Mat 轉換為 torch::Tensor 並將其提供給 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);
}