pytorch/vision

windows maskrcnn error: unresolved external symbol "class at::Tensor __cdecl nms_cpu(class at::Tensor const &,class at::Tensor const &,float)" (?nms_cpu@@YA?AVTensor@at@@AEBV12@0M@Z)

開放

#2,139 建立於 2020年4月24日

 (22 則留言) (0 個反應) (0 位負責人)Python (6,858 個分叉)batch import
help wantedmodule: c++ frontendmodule: opswindows

倉庫指標

星標
 (15,050 顆星)
PR 合併指標
 (PR 指標待抓取)

描述

I built torchvision pulled from master using :

cmake -DCMAKE_PREFIX_PATH=D:/Garage/onnx-deploy/libtorchd  -DCMAKE_INSTALL_PREFIX=D:/vision/Debug -DCMAKE_BUILD_TYPE=Debug ..
cmake --build . --config Debug 
cmake --install .

Then torch.jit.script(model) and saved mask-rcnn model from torchvision. Errors occured in my C++ project, firstly Unknown builtin op: torchvision::_new_empty_tensor_op, then I modified my code according to #1407, added lines like #include "torchvision/nms.h" and torch::RegisterOperators().op("torchvision::nms", &nms) , but it won't work. The error became unresolved external symbol "class at::Tensor __cdecl nms_cpu(class at::Tensor const &,class at::Tensor const &,float)" (?nms_cpu@@YA?AVTensor@at@@AEBV12@0M@Z). The libtorch version is 1.5.0 and torchvision is pulled from master yesterday. Any clue how I can get mask rcnn running? Similar issues are many but little to help, I really get confused about how to modify CMakeList.txt of TorchVision.

cc @peterjc123 @nbcsm @guyang3532 @maxluk @gunandrose4u @smartcat2010 @mszhanyi

貢獻者指南