SJTU-Thinklab-Det / r3det-pytorch

R3Det based on mmdet 2.19.0
Apache License 2.0
42 stars 7 forks source link

Merge时在obb_nms遇到bug应该怎么解决? #1

Closed Virusxxxxxxx closed 2 years ago

Virusxxxxxxx commented 2 years ago

当我调用test.py测试rretinanet_obb_r50_fpn_1x_dota_v3时,总是在Merging patch这一步卡住,经过调试我发现是在nms_rotated_wrapper.py的40行出的问题,上面的注释也写了可能会出现问题,请问我怎么解决这个bug?万分感谢。

Virusxxxxxxx commented 2 years ago

问题已解决。 解决方法:r3det/datasets/dota1.py的247行,我把4改成了1。

顺便记录一下test v2遇到一个问题RuntimeError: Cannot re-initialize CUDA in forked subprocess. 解决方法:在test.py调用main()之前,加一句torch.multiprocessing.set_start_method('spawn')

zytx121 commented 2 years ago

Thanks for your question! It looks like you have solved it by yourself. We will fix this bug in r3det/datasets/dota1.py.