jbwang1997 / OBBDetection

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

多卡disttest出现报错 #23

Closed zhaogev5 closed 2 years ago

zhaogev5 commented 2 years ago

当我使用多次卡测试时出现了如下报错,请问是什么原因呢 obb_anchor_head.py", line 585, in get_bboxes
img_shape = img_metas[img_id]['img_shape'] TypeError: 'DataContainer' object is not subscriptable 感谢您

jbwang1997 commented 2 years ago

这好像是pytorch某一个版本的bug,可以换一下pytorch版本试一下