时间戳

首页 > 解决方案 > Libtorch C++ - 没有匹配的成员函数来调用 InterpolateFuncOptions 的“大小”

c++ - Libtorch C++ - 没有匹配的成员函数来调用 InterpolateFuncOptions 的“大小”

问题描述

在 C++ 中使用 Libtorch 1.6.0，我收到以下错误：

error: no matching member function for call to 'size'

我的线路如下：

image = F::interpolate(image, F::InterpolateFuncOptions().size({target_height, target_width}).mode(torch::kNearest));

但在文档中它似乎是正确的......有什么想法吗？

提前致谢

标签： c++pytorchlibtorch

解决方案

你应该这样包装它std::vector：

image = F::interpolate(image, 
        F::InterpolateFuncOptions()
        .size(std::vector<>{target_height, target_width})
        .mode(torch::kNearest));

原因是size没有对您尝试使用的std::initializer_list的重载调用（请参阅此处size的文档）。

推荐阅读