ming71 / rotate-yolov3

Arbitrary oriented object detection implemented with yolov3 (attached with some tricks).
223 stars 58 forks source link

There is no a r_nms script (ModuleNotFoundError: No module named 'utils.nms.r_nms') #9

Closed jingweirobot closed 4 years ago

jingweirobot commented 4 years ago

There is no a r_nms script. So cannot go through this package. Where is the r_nms script.

parser.add_argument('--cfg', type=str, default='cfg/ICDAR/yolov3_608_dh_o8_ga.cfg', help='cfg file path') parser.add_argument('--data', type=str, default='data/tiny.data', help='*.data file path')

But if I want to test HRSC dataset, which one can I choose, yolov3-416.cfg or yolov3-m.cfg? thanks in advance.

ming71 commented 4 years ago

Hi, you need to compile cuda file first, and then import r_nms function, more details can be founded in utils/nms. Try yolov3-416.cfg for HRSC dataset. yolov3-m.cfg is built for matrixent(ICCV 2019), but didn't performed well, thus not recommended.

jingweirobot commented 4 years ago

Ok. sure. I cannot find the corresponding cuda package in rotate-yolov3. But there is a cuda package uploaded on March 16th in your repo, I found. It is available for rotate-yolov3? Can I use this directly for compiling? thanks

ming71 commented 4 years ago

Yes, I've committed rnms codes to this repo, it's the same with implementation in ryolo, good luck.

jingweirobot commented 4 years ago

Hi. I still cannot compile CUDA RNMS GPU successfully. Where could I find the original package (RNMS). thanks

ming71 commented 4 years ago

This is the original code I referred to. Though I failed to run his codes, RNMS part works well and I've checked.

jingweirobot commented 4 years ago

Hi. actually, I received an error still while compiling RNMS. downgraded gcc-7(it is not ok) to gcc-5. the terminal shows:

Installed /home/user/rotate-yolov3/utils/nms Processing dependencies for r-nms==0.0.0 Finished processing dependencies for r-nms==0.0.0

But another issue likes this. ImportError: /home/user/rotate-yolov3/utils/nms/r_nms.cpython-37m-x86_64-linux-gnu.so: undefined symbol: cudaSetupArgument.

By the way, could you provide the corresponding paltform, I mean, such as cuda version, gcc version, something like this.

thanks

jingweirobot commented 4 years ago

Hi could you directly upload or send the r_nms function or r_nms script to me ? thanks

ming71 commented 4 years ago

I tested just now, and Ubuntu 16.04, cuda 10.0, gcc 7.4 is fine for me: res

jingweirobot commented 4 years ago

Thanks. Could you share the screen regarding compiling CUDA rnms? or that package after compiling. Let me check what happens about this issue on my platform.

ming71 commented 4 years ago

This is the compiling log: compile and the root dir ( installation is not necessary): rootdir

satpalsr commented 2 years ago

Hey @jingweirobot, were you able to work it out? Facing the same rnms problem.