lulersoft

Me too,

删除 #include "torchvision/nms.h" 在src_v4.cpp 加入以下： ```` template at::Tensor nms_cpu_kernel( const at::Tensor& dets, const at::Tensor& scores, const float iou_threshold) { AT_ASSERTM(!dets.options().device().is_cuda(), "dets must be a CPU tensor"); AT_ASSERTM( !scores.options().device().is_cuda(), "scores must...

这是俺项目的 CMakeLists.txt，仅供参考，win10,vs2019 编译通过 ```` # CMakeList.txt: torch_exe 的 CMake 项目，在此处包括源代码并定义 # 项目特定的逻辑。 # cmake_minimum_required (VERSION 3.8) project(torch_exe) set(CMAKE_PREFIX_PATH "D:\\sdk\\libtorch\\lib") #//注意这里填自己解压libtorch时的路径 set(OpenCV_DIR "D:\\sdk\\opencv\\build") find_package(Torch REQUIRED) find_package(OpenCV QUIET) if(NOT Torch_FOUND) message(FATAL_ERROR "Pytorch...