search

jbwang1997 / OBBDetection

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

你好，如果我想把检测时不相关的类别屏蔽掉，这步怎么弄呢 #140

Open ustbwanglu opened 2 years ago

ustbwanglu commented 2 years ago

看到您之前问过类似问题，能否有个详细的解决办法

jbwang1997 commented 2 years ago

模型输出的是一个列表，可以将其他类别的 ndarray 都设置为空就过滤掉了。