libtorch使用中遇到的一些坑
由 hubdog 于 星期三, 2024-04-10 07:36 发表
libtorch是个C++的库,不像tensorflow那样提供了C的api,用起来有点不方便
自己去下载的库有很多问题,执行过程中会crash,我后来使用的下面这个网站下载的库
https://pytorch.org/get-started/locally/
图像转换这块,libtorch没有提供numpy的接口,所以,要用opencv读进来,然后用from_blob去转换,这里面还有几个坑。以后有空会说。
创建的Tensor一定要注意不仅是Shape要跟模型要求的一致,而且数据类型也要一致,否则出力结果会莫名其妙。使用模型推导,可以直接Model(input), 也可以用Model.forward。两者是等价的。