search

zzzxxxttt / pytorch_simple_CenterNet_45

A simple pytorch implementation of CenterNet (Objects as Points)
308 stars 63 forks source link

There is no nms folder in the repo #13

Closed songyuc closed 4 years ago

songyuc commented 4 years ago

Hi, guys, I am continuing learning this repo today, and I find there is no nms folder in this repo, which has been imported in the test.py, as 图片 So, how can I fix this?

Your answer and idea will be appreciated!

zzzxxxttt commented 4 years ago

Just follow the README to install nms and run the code, I think most of your questions can be answered if you run it before opening a new issue.

songyuc commented 4 years ago

Hi, @zzzxxxttt , I am really sorry about my absent knowledge and experience, In fact, I am just confused, as,

  1. there no nms folder in the repo like, 图片
  2. nms is in the libfolder as, cd $CenterNet_ROOT/lib/nms so how does 'from nms.nms import soft_nms' work, or maybe 'from lib.nms.nms import soft_nms' would work?
zzzxxxttt commented 4 years ago

After installing nms, it is located in your python package folder, just like you don't need a "torch" folder in your project.

songyuc commented 4 years ago

After looking for more information, I have known that it is a drawback within the PyCharm. But this might not influence the normal running of this repo. Thanks sincerely again for the guide from @zzzxxxttt.