search

microsoft / CNTK

Microsoft Cognitive Toolkit (CNTK), an open source deep-learning toolkit
https://docs.microsoft.com/cognitive-toolkit/
Other
17.53k stars 4.28k forks source link

cntk provide Faster_Rcnn nms function for C#? #3361

Open lotuschang opened 6 years ago

lotuschang commented 6 years ago

Hi, I can load Faster_Rcnn model with C#, and get outputdata from Evaluate. I got clsPred/rois/vbboxR list data and reference python code to generate input for NMS. cntk provide nms function to use for C#? python can import utils.cython_modules.cpu_nms. C# has the same way? thanks for your help.

ke1337 commented 6 years ago

Please check the doc for NMS library source. There are several ways to do it:

  1. pyd files are actually just dll in Windows. You may use P/Invoke to call function directly
  2. You may compile a C++ nms lib from cython then, according to this, and then wrap it up in C#
  3. Since both 1 and 2 would have dependency on python dlls, a better solution might be writing a native C++ implementation for NMS. We have it in our backlog.