【SLAM】——GCNv2-SLAM修改成CPU版本

GCNv2, SLAM
GPU版本的编译可参考:编译GCNv2_SLAM

  1. GCNextractor.cc的218行改为:
    torch::DeviceType device_type;
    device_type = torch::kCPU;
    torch::Device device(device_type);

    const char *net_fn = getenv("GCN_PATH");
    net_fn = (net_fn == nullptr) ? "gcn2.pt" : net_fn;
    module = torch::jit::load(net_fn, device);
  1. 使用全局搜索,将torch::kCUDA替换为torch::kCPU

  2. GCN2/gcn2_320x240.pt文件中的gcn/code/gcn.py中的cuda:0替换为cpu

  3. 跑起来的效果
    在这里插入图片描述

评论 8
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值