Linux下libtorch导入VTM编码器

1. libtorch下载

此处注意下载和pytorch版本一致的libtorch

wget https://download.pytorch.org/libtorch/cpu/libtorch-cxx11-abi-shared-with-deps-2.0.1%2Bcpu.zip
unzip 下载的libtorch压缩包

2. 修改CMakeLists.txt

• 首先在VTM编码器的文件夹下的CMakeLists.txt中，添加相对当前vtm文件夹路径的 libtorch头文件，库路径，如图
  
  然后修改此CMakeLists.txt中的c++编译器版本至c++14，如图
  
  然后注释此CMakeLists.txt中的警告，如图
  
• 继续修改VTM中目录VTM/source/App/EncoderApp/ 下的CMakeLists.txt文件，给EncoderApp添加编译库路径，也就是把libtorch/lib/下的库添加进去，为了保证正常运行我就全添加了，如图，
  

3. 注释或者修改libtorch中与VTM编码器中的重命名

在目录~libtorch\include\c10\util下的logging_is_not_google_glog.h 头文件中

注释掉重命名 CHECK

119 // Check for a given boolean condition.
120 // #define CHECK(condition) FATAL_IF(condition) << "Check failed: " #condition " "

4. 运行

可能运行还是会出现问题，多半是libtorch库路径的问题，把相关路径配置到用户目录下的 .bashrc文件下，如图

保险起见也可以在 /etc/profile文件下面配置此路径，并且把libtorch的库包复制到/usr/lib/ 目录下

本文来自互联网用户投稿，文章观点仅代表作者本人，不代表本站立场，不承担相关法律责任。如若转载，请注明出处。 如若内容造成侵权/违法违规/事实不符，请点击【内容举报】进行投诉反馈！