search

jbwang1997 / OBBDetection

OBBDetection is an oriented object detection library, which is based on MMdetection.
Apache License 2.0
522 stars 112 forks source link

huge image测试问题反馈 #86

Open whystopped opened 2 years ago

whystopped commented 2 years ago
  1. 请问BT_nms是什么，我更换成obb_nms（根据代码应该是有gpu加速）后，nms速度快了不少
  2. mmdet/models/detectors/obb/obb_base.py中img = img.copy()一行（第24行）消耗了很多运行时间，由于我的图片较大（15000*15000），程序会在这里卡住
  3. BboxToolkit/BboxToolkit/visualization/show.py由于采用pyplot，同样有上述卡住的问题，暂时不知道怎么解决
jbwang1997 commented 2 years ago

BT_nms是使用BboxToolkit中的nms，允许多种形式旋转框的nms，这里主要是为了照顾POLY输出形式 img.copy()是防止对原本的ndarray做出改变，show_result也可以输入ndarray的格式。 前两点可以根据需要进行修改。

对于第三点我会尽量增加cv2可视化的后台，或者可以尝试直接保存下来。

whystopped commented 2 years ago

BT_nms是使用BboxToolkit中的nms，允许多种形式旋转框的nms，这里主要是为了照顾POLY输出形式 img.copy()是防止对原本的ndarray做出改变，show_result也可以输入ndarray的格式。 前两点可以根据需要进行修改。

对于第三点我会尽量增加cv2可视化的后台，或者可以尝试直接保存下来。

感谢！

whystopped commented 2 years ago

抱歉，与我这边的服务器疑似被挖矿，导致CPU占用率过高有关

whystopped commented 2 years ago

抱歉，与我这边的服务器疑似被挖矿，导致CPU占用率过高有关

不行，还是有时候会卡在BboxToolkit/BboxToolkit/visualization/show.py第82行plt.imshow(img)

我暂时没搞明白在什么情况下会卡住

jbwang1997 commented 2 years ago

cv2不会有这样的问题是吗

whystopped commented 2 years ago

cv2不会有这样的问题是吗

是的，cv2不会这样，我后来在其他服务器上试过，pyplot虽然更慢但是可以正常运行，而我这边处于偶尔正常偶尔卡住的情况，我觉得如果其他人没有类似问题的话，可能还是我本地的问题