编译
cmake -DCMAKE_PREFIX_PATH="/home/amax/download/libtorch" -DCMAKE_C_COMPILER=/usr/bin/gcc -DCMAKE_CXX_COMPILER=/usr/bin/g++ ..
libtorch-yolov3/Darknet.cpp
// at::Tensor weights = torch::CPU(torch::kFloat32).tensorFromBlob(weights_src, {length/4});
527 at::Tensor weights = torch::from_blob(weights_src,{length/4}).to(torch::kCUDA);
570 bn_running_var = bn_running_var.view_as(bn_imp->running_var);
bn_imp->running_var.set_data(bn_running_var);
main.cpp:60:51
// auto img_tensor = torch::CPU(torch::kFloat32).tensorFromBlob(img_float.data, {1, input_image_size, input_image_size, 3});
auto img_tensor = torch::from_blob(img_float.data, {1, input_image_size, input_image_size, 3}).permute({0, 3, 1, 2}).to(torch::kCUDA);
img_tensor = img_tensor.permute({0,3,1,2});
auto img_var = torch::autograd::make_variable(img_tensor, false).to(device);
以上的错误 地方做了一些修改,最后发现还是会有问题,
自己做了个修正,放在github上,改成了usb摄像头的版本,欢迎拍砖