GCNv2, SLAM
GPU版本的编译可参考:编译GCNv2_SLAM
- 将
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);
-
使用全局搜索,将
torch::kCUDA
替换为torch::kCPU
-
将
GCN2/gcn2_320x240.pt
文件中的gcn/code/gcn.py
中的cuda:0替换为cpu -
跑起来的效果