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 opened on 2020/04/24

 (22 件のコメント) (0 件のリアクション) (0 人の担当者)Python (6,858 件のフォーク)batch import
help wantedmodule: c++ frontendmodule: opswindows

Repository metrics

Stars
 (15,050 個のスター)
PR merge metrics
 (PR metrics pending)

説明

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

コントリビューターガイド